leetcode 113. 路径总和 II

2019-04-10  本文已影响0人  topshi

题目描述

给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
相关话题:树、深度优先搜索    难度:中等
说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树,以及目标和 sum = 22,


返回:

[
[5,4,11,2],
[5,8,4,5]
]

思路:这道题和112题很像,唯一不同是它需要把路径保存下来。
那么我们就需要想办法把遍历过的路径保存下来。

public List<List<Integer>> pathSum(TreeNode root, int sum) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        List<Integer> list = new ArrayList<Integer>();
        pathSum(root,sum,res,list);
        return res;
    }
    public void pathSum(TreeNode root, int sum, List<List<Integer>> res, List<Integer> list){
        if(root == null) return;
        list.add(root.val);
        if(root.left == null && root.right == null && sum == root.val){
            res.add(new ArrayList(list));
        }
        pathSum(root.left,sum-root.val, res, list);
        pathSum(root.right,sum-root.val, res, list);
        list.remove(list.size()-1);
    }
上一篇 下一篇

猜你喜欢

热点阅读