leecode 113. 路径总和 II

113. 路径总和 II

leecode 113. 路径总和 II_第1张图片

        这题是112. 路径总和 的变体,需要注意两个题的区别,112题是求二叉树是否存在一个与目标值相等的和,如果有相等的值,直接返回,不用管其他分支是否还有,因为这个树上已经存在一个与目标值相等的分支,因此这题不用遍历整个树;而本题是需要找到所有与目标值相等的分支,因此需要对整个树遍历,当遍历到符合目标值的时候,就将这个分支记录下来,直到整棵树遍历后再去看有多少结果,最终输出所有符合要求的树分支。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
//这里用了List,也可以用Deque,Deque可以用来做队列和栈使用
    List> res = new LinkedList<>();
    Deque path = new LinkedList();
    public List> pathSum(TreeNode root, int targetSum) {
        dfs(root,targetSum);
        return res;
    }

    private void dfs(TreeNode root, int targetSum) {
        if (root == null) return;
//这里不管该节点是否最后符合结果,都加入path,如果最终targetSum不符合,那么不加入res即可
        path.offer(root.val);
        targetSum -= root.val;
        if (root.left == null && root.right == null && targetSum == 0) {
            res.add(new LinkedList(path));
        }
        dfs(root.left, targetSum);
        dfs(root.right, targetSum);
//需要删除当前path中最后一个值,因为经过上面遍历后这个值已经不符合结果了,
//需要回溯从上一个值的右节点继续找
        path.pollLast();
    }
}

        这里有一个问题,最后回溯的那一步:path.pollLast(); 为什么dfs递归了左右子树,这里只回溯了一次?
        因为这里回溯的是当前节点,打断点可以看到前面的dfs已经把子树的值移出了。只有遍历到叶子节点的时候才会把最后加入的叶子节点弹出栈。已经使用,没有其他价值。只有是叶子节点的时候才需要回溯,左右子树只有一个为空返回的话是不需要回溯的。

        pop是pop的 path.offer(root.val);

        这里的poll,其实是移出的当前的节点,而移出当前节点的前提是已经遍历了当前节点的左右子树,这里的dfs已经遍历了当前节点的左右子树,所以这里不矛盾。

你可能感兴趣的:(二叉树,回顾,数据结构,算法,java,leetcode,二叉树)