题解 | #二叉树根节点到叶子节点和为指定值的路径#

二叉树根节点到叶子节点和为指定值的路径

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);
    }
}
全部评论

相关推荐

10-25 12:05
已编辑
湖南科技大学 Java
若梦难了:我有你这简历,已经大厂乱杀了
点赞 评论 收藏
分享
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务