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

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

http://www.nowcoder.com/practice/840dd2dc4fbd4b2199cd48f2dadf930a

标准的递归回溯算法,题目小坑,又不说有没有负数
其实还可以广度遍历,记录每一条路径并记录其长度--或者最后算长度

int cur = 0;
    public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
        // write code here
        //深度遍历,回溯算法
        ArrayList<Integer> con = new ArrayList<>();
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        DFS(root,con,sum,res);
        return res;
    }

    private void DFS(TreeNode root, ArrayList<Integer> con, int sum, ArrayList<ArrayList<Integer>> res) {
        if(root==null){
            return;
        }
        cur+=root.val;
        con.add(root.val);
        if (cur==sum&&root.left==null&&root.right==null){
            res.add(new ArrayList<>(con));
            //回退
            con.remove(con.size()-1);
            cur-=root.val;
            return;
        }
        DFS(root.left,con,sum,res);
        DFS(root.right,con,sum,res);
        //回退
        con.remove(con.size()-1);
        cur-=root.val;
    }
全部评论

相关推荐

老方子:英语等级cet写错了吧
点赞 评论 收藏
分享
数学转码崽:一直给我推,投了又不理,理了又秒挂
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务