LCR 050-路径总和 III

Raphael Liu Lv10

给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum路径 的数目。

路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。

示例 1:

**输入:** root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
**输出:** 3
**解释:** 和等于 8 的路径有 3 条,如图所示。

示例 2:

**输入:** root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
**输出:** 3

提示:

  • 二叉树的节点个数的范围是 [0,1000]
  • -109 <= Node.val <= 109
  • -1000 <= targetSum <= 1000

注意:本题与主站 437 题相同:https://leetcode-cn.com/problems/path-sum-iii/

暴力枚举每个结点作为根,在对枚举的每颗子树进行dfs,判断是否存在targetSum。
注意:过程sum可能会溢出,故设为long。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
class Solution {
long target;
int ans = 0;

public void dfs(TreeNode u, long sum) {
if (u == null) return;
if (sum + u.val == target){
ans++;
// System.out.println(sum + " " + u.val);
}
dfs(u.left, sum + u.val);
dfs(u.right, sum + u.val);
}

public void dfs1(TreeNode u){
if(u == null) return;
dfs(u, 0);
// System.out.println(u.val + " " + ans + "\n");
dfs1(u.left);
dfs1(u.right);

}
public int pathSum(TreeNode root, int targetSum) {
this.target = targetSum;
dfs1(root);
return ans;
}
}
 Comments
On this page
LCR 050-路径总和 III