文章目录
题目描述
给定一个二叉树和一个值 sum,判断 是否有从根节点到叶子节点的节点值之和等于 sum 的路径,
例如:
给出如下的二叉树,sum=22,
返回true,因为存在一条路径 5→4→11→2的节点值之和为 22
示例1
输入
{1,2},0
输出
false
示例2
输入
{1,2},3
输出
true
题解思路
递归
实现代码:
bool hasPathSum(TreeNode* root, int sum) {
if (nullptr == root) return false;
else if (root->val == sum && root->left == nullptr && root->right == nullptr) {
return true;
}
return hasPathSum(root->left, sum - root->val) || hasPathSum(root->right, sum - root->val);
}
题目描述
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。 从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
题解思路
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
vector<vector<int>> res;
vector<int> cur;
dfs(root, sum, res, cur);
return res;
}
void dfs(TreeNode* root, int sum, vector<vector<int>>& res, vector<int>& cur){
if(!root) return;
cur.push_back(root->val);
sum -= root->val;
if(sum == 0 && !root->left && !root->right) res.push_back(cur); // 满足路径条件
dfs(root->left, sum, res, cur);
dfs(root->right, sum, res, cur);
cur.pop_back(); // 关键点:回溯
}
};
如有不同见解,欢迎留言讨论~~~