天天看点

剑指offer JZ24 二叉树中和为某一值的路径

题目链接:

JZ24 二叉树中和为某一值的路径

本题思路:

public class Solution {
    // 结果列表res中可能有多个列表
    // 所以ArrayList<ArrayList<Integer>>
    ArrayList<ArrayList<Integer>> res = new ArrayList<>(); // 输出结果
    ArrayList<Integer> path = new ArrayList<>(); // 暂存路径
    
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        // 注意:路径为 根结点一直到叶结点 所经过的结点
        if(root == null || root.val > target) return res;
        path.add(root.val);
        target -= root.val;
        
        // 到叶结点且和为target
        if(target == 0 && root.left == null && root.right == null)
            // 把path列表复制到结果列表
            res.add(new ArrayList<Integer>(path));
        
        // 左右子树递归查找
        FindPath(root.left,target);
        FindPath(root.right,target);
        // 到叶子节点仍不满足条件 退回上一节点
        path.remove(path.size() - 1);
        
        return res;
    }
}
           

继续阅读