重构决策树的代码