题解 | #二叉树根节点到叶子节点和为指定值的路径#
二叉树根节点到叶子节点和为指定值的路径
http://www.nowcoder.com/practice/840dd2dc4fbd4b2199cd48f2dadf930a
此文仅用于本人学习记录
可算是给我过了,这里需要注意的是,二维数组使用add方法添加一维数组必须用new ArrayList<>(temp)而不是直接add(temp)。还有关键的一点就是防止无效添加,如果只使用一个temp来记录当前路径的话,必须返回前除去添加的值。成功达到value的时候就直接添加了所以不用怕temp会缺少值。
记住二叉树递归是可以返回父节点的,就是函数结束的那一刻,继续执行父节点在递归之后的操作。
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* }
*/
public class Solution {
/**
*
* @param root TreeNode类
* @param sum int整型
* @return int整型ArrayList<ArrayList<>>
*/
public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
// write code here
if(root == null) return new ArrayList<>();
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
dfs(root,sum,0,new ArrayList<Integer>(),result);
return result;
}
public void dfs(TreeNode root,int sum,int value,ArrayList<Integer> temp,ArrayList<ArrayList<Integer>> result){
temp.add(root.val);
value += root.val;
if(root.left == null && root.right == null){
if(value == sum){
result.add(new ArrayList<>(temp));
}
}
if(root.left != null){
dfs(root.left,sum,value,temp,result);
}
if(root.right != null){
dfs(root.right,sum,value,temp,result);
}
temp.remove(temp.size() - 1);
}
}