112 路径总和

本文最后更新于:2021年7月22日 晚上

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ,判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum

叶子节点 是指没有子节点的节点。

示例 1:

img

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

示例 2:

img

1
2
输入:root = [1,2,3], targetSum = 5
输出:false

示例 3:

1
2
输入:root = [1,2], targetSum = 0
输出:false

提示:

  • 树中节点的数目在范围 [0, 5000]
  • -1000 <= Node.val <= 1000
  • -1000 <= targetSum <= 1000

Solution

  • 递归法
  • 「如果需要搜索整颗二叉树,那么递归函数就不要返回值,如果要搜索其中一条符合条件的路径,递归函数就需要返回值,因为遇到符合条件的路径了就要及时返回。」
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
29
30
31
32
33
// @lc code=start
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
bool hasPathSum(TreeNode* root, int targetSum) {
if (root == nullptr) return false;
return traversal(root, targetSum - root->val);
}
private:
bool traversal(TreeNode* root, int sum) {
if (!root->left && !root->right && sum==0) return true;
if (!root->left && !root->right) return false;

if (root->left) {
if (traversal(root->left, sum - root->left->val)) return true;
}
if (root->right) {
if (traversal(root->right, sum - root->right->val)) return true;
}
return false;
}
};
// @lc code=end

简化后

1
2
3
4
5
6
7
8
9
10
11
class Solution {
public:
bool hasPathSum(TreeNode* root, int targetSum) {
if(root==nullptr)
return false;
if(root->left==nullptr && root->right==nullptr)
return root->val==targetSum;
return hasPathSum(root->left, targetSum-root->val) ||
hasPathSum(root->right, targetSum-root->val);
}
};
  • 广度优先搜索,参考 LeetCode官方
  • 使用两个队列,分别存储将要遍历的节点,以及根节点到这些节点的路径和即可。
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
29
30
31
class Solution {
public:
bool hasPathSum(TreeNode* root, int targetSum) {
if(root==nullptr)
return false;

queue<TreeNode *> qNode;
queue<int> qSum;
qNode.push(root);
qSum.push(root->val);

while(!qNode.empty()){
TreeNode* cur = qNode.front();
int sum = qSum.front();
qNode.pop(); qSum.pop();
if(cur->left==nullptr && cur->right==nullptr){
if(sum == targetSum) return true;
continue;
}
if(cur->left != nullptr){
qNode.push(cur->left);
qSum.push(cur->left->val+sum);
}
if(cur->right != nullptr){
qNode.push(cur->right);
qSum.push(cur->right->val+sum);
}
}
return false;
}
};

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!