Nowcoder java-二叉树中和为某一值的路径
(题目来自牛客网)
思路
参考牛客网上的同学的思路分析:
本题要求找到从根节点到各个叶子节点的每一条路径,并把路径上的所有节点之和与某个数相比较,这符合深度优先遍历(DFS)的特点。
参考了几位同学的做法,主要方法useDFS的思路如下:
一、将传入方法的节点的值加入路径列表path。
二、判断该节点是否有孩子节点。若没有孩子,则该节点为叶子节点,表示该路径已走完。比较该节点值和目标数值target是否相同,相同则把路径列表path添加到结果列表list中。 若有孩子,表示该路径还没走完。开始递归: 1.左孩子不为空,将左孩子节点以及新的目标数值(target-该节点值)作为参数传入useDFS,实现对左孩子的深度优先遍历。 2.右孩子不为空,进行相同的操作。
三、以上几步均完成后,表示本次方法中的节点及其左右孩子都已经被遍历完毕,也就是说经过本节点的路径已经统计完毕。此时对路径列表list进行回溯,移除其最后一个元素,结束方法。
注:将路径列表path添加到结果列表list中时,不要直接list.add(path),这样添加的所有元素都是同一个path的引用,是错的。此处应为 list.add(new ArrayList(path));
代码
import java.util.ArrayList;
public class FindPath {
ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> path = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> findPath(TreeNode root, int target) {
if (root == null){
return list;
}
path.add(root.val);//因为FindPath中和 下面程序中都进行了判null操作,root绝对不可能为 null
target -= root.val;
if (target == 0&& root.left == null && root.right == null){//已经到达叶子节点,并且正好加出了target
list.add(new ArrayList<Integer>(path));//将该路径加入list结果集中
}
findPath(root.left,target);//如果左子树非空,递归左子树
findPath(root.right,target);//如果右子树非空,递归右子树
path.remove(path.size() -1); //无论当前路径是否加出了target,必须去掉最后一个,然后返回父节点,去查找另一条路径,最终的path肯定为null
return list;
}
public static void main(String[] args) {
TreeNode node_10 = new TreeNode(10);
TreeNode node_5 = new TreeNode(5);
TreeNode node_12 = new TreeNode(12);
TreeNode node_4 = new TreeNode(4);
TreeNode node_7 = new TreeNode(7);
node_10.left = node_5;
node_10.right = node_12;
node_5.left = node_4;
node_5.right = node_7;
System.out.println(new FindPath().findPath(node_10, 22));
}
}