题解 | #二叉树根节点到叶子节点和为指定值的路径#
二叉树根节点到叶子节点和为指定值的路径
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); } }