ID: 2602

Difficulty: 5

所属公司 : 阿里

时间 :2025年5月18日-菜鸟(算法岗)

算法标签>

CART决策树

题目内容

给定一个数据集，其中每一行代表一个样本，每一列代表一个特征，最后一列是样本的标签。请你计算每个特征的基尼指数，并输出基尼指数最小的特征的索引和对应的基尼指数。基尼指数的计算公式:对于一个特征，其基尼指数定义为:

题解思路与方法

问题分析

给定一个二维数据集，每行是一个样本，最后一列为标签。要对每个特征计算基尼指数，选择基尼值最小的特征。基尼指数常用于决策树（CART）算法的划分标准。

算法流程

记样本总数为 $n$ ，特征数为 $m$ 。