原题目
面试题 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)
小结
-
第一次遇见前缀和,应再回顾一下网上这个解法。
-
对容器的使用方法不是很熟练,下一篇博客要总结一下常用容器的使用方法。