剑指 Offer 34. 二叉树中和为某一值的路径
题目内容
给你二叉树的根节点 root 和一个整数目标和 targetSum ，找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
示例 1：
1
2
| 输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：[[5,4,11,2],[5,8,4,5]]
|
示例 2：
1
2
| 输入：root = [1,2,3], targetSum = 5
输出：[]
|
示例 3：
1
2
| 输入：root = [1,2], targetSum = 0
输出：[]
|
提示：
1
2
3
| 树中节点总数在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
|
解法一：深度优先搜索
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
34
35
36
37
38
39
| /**
* 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 {
private:
int targetNum;
vector<vector<int>> res;
public:
void dfs(TreeNode* root,int& parentNum,vector<int> list){
if(!root) return;
int current = root->val + parentNum;
list.push_back(root->val);
if(current == targetNum && (!root->left) && (!root->right)){
res.push_back(list);
}
dfs(root->left,current,list);
dfs(root->right,current,list);
}
vector<vector<int>> pathSum(TreeNode* root, int target) {
if(!root) return {};
if(target == root->val && (!root->left) && (!root->right))return {{root->val}};
targetNum = target;
dfs(root->left,root->val,{root->val});
dfs(root->right,root->val,{root->val});
return res;
}
};
|
结果
1
2
3
4
5
6
7
8
9
10
11
12
| 执行用时：
16 ms
, 在所有 C++ 提交中击败了
16.27%
的用户
内存消耗：
37.9 MB
, 在所有 C++ 提交中击败了
5.01%
的用户
通过测试用例：
114 / 114
|
