04.12 求和路径

原题目

面试题 04.12. 求和路径

给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

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

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

3
解释：和为 22 的路径有：[5,4,11,2], [5,8,4,5], [4,11,7]

提示：

• 节点总数 <= 10000

第一遍解法

暴力法，遍历每个节点，对于每个节点，计算三个路径数量之和。

• 以此节点为根节点的路径之和为target的数量。
• 左子树下路径之和为target的数量。
• 右子树下路径之和为target的数量。

class Solution {
    public int pathSum(TreeNode root, int sum) {
        if (root == null) {
            return 0;
        }
        int pathsFromRoot = pathSum(root, sum, 0);
        int pathsFromLeft = pathSum(root.left, sum);
        int pathsFromRight = pathSum(root.right, sum);

        return pathsFromRoot + pathsFromLeft + pathsFromRight;
    }

    public int pathSum(TreeNode root, int sum, int curSum) {
        if (root == null) {
            return 0;
        }

        curSum += root.val;
        
        int cnt = 0;
        if (curSum == sum) {
            cnt++;
        }

        cnt += pathSum(root.left, sum, curSum);
        cnt += pathSum(root.right, sum, curSum);

        return cnt;
    }
}

时间复杂度

• 假设树是平衡的，那么高度为nlog(n)，在根节点处，遍历下面的N-1个节点，下一层遍历N-3，这样

  • (N - 1) + (N - 3) + (N - 7) + … + (N - N) = O(N * log(N))

• 假设树是线性的，那么高度为n，这样

  • (N - 1) + (N - 2) + (N - 3) + … + (N - N) = O(N ^ 2)

网上好的解法

前缀和，因为自己也是模模糊糊理解了，所以不知道怎么写自己的思想，后面再回顾吧。

思想参考原作者：04.12 求和路径

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int pathSum(TreeNode root, int sum) {
        // key是前缀和, value是大小为key的前缀和出现的次数
        Map<Integer, Integer> prefixSumCount = new HashMap<>();
        // 前缀和为0的一条路径
        prefixSumCount.put(0, 1);
        // 前缀和的递归回溯思路
        return recursionPathSum(root, prefixSumCount, sum, 0);
    }

    /**
     * 前缀和的递归回溯思路
     * 从当前节点反推到根节点(反推比较好理解，正向其实也只有一条)，有且仅有一条路径，因为这是一棵树
     * 如果此前有和为currSum-target,而当前的和又为currSum,两者的差就肯定为target了
     * 所以前缀和对于当前路径来说是唯一的，当前记录的前缀和，在回溯结束，回到本层时去除，保证其不影响其他分支的结果
     * @param node 树节点
     * @param prefixSumCount 前缀和Map
     * @param target 目标值
     * @param currSum 当前路径和
     * @return 满足题意的解
     */
    private int recursionPathSum(TreeNode node, Map<Integer, Integer> prefixSumCount, int target, int currSum) {
        // 1.递归终止条件
        if (node == null) {
            return 0;
        }
        // 2.本层要做的事情
        int res = 0;
        // 当前路径上的和
        currSum += node.val;

        //---核心代码
        // 看看root到当前节点这条路上是否存在节点前缀和加target为currSum的路径
        // 当前节点->root节点反推，有且仅有一条路径，如果此前有和为currSum-target,而当前的和又为currSum,两者的差就肯定为target了
        // currSum-target相当于找路径的起点，起点的sum+target=currSum，当前点到起点的距离就是target
        res += prefixSumCount.getOrDefault(currSum - target, 0);
        // 更新路径上当前节点前缀和的个数
        prefixSumCount.put(currSum, prefixSumCount.getOrDefault(currSum, 0) + 1);
        //---核心代码

        // 3.进入下一层
        res += recursionPathSum(node.left, prefixSumCount, target, currSum);
        res += recursionPathSum(node.right, prefixSumCount, target, currSum);

        // 4.回到本层，恢复状态，去除当前节点的前缀和数量
        prefixSumCount.put(currSum, prefixSumCount.get(currSum) - 1);
        return res;
    }
}

作者：burning-summer
链接：https://leetcode-cn.com/problems/paths-with-sum-lcci/solution/qian-zhui-he-de-ying-yong-di-gui-hui-su-by-shi-huo/
来源：力扣（LeetCode）

时间复杂度: O(n)

小结

• 第一次遇见前缀和，应再回顾一下网上这个解法。

• 对容器的使用方法不是很熟练，下一篇博客要总结一下常用容器的使用方法。