二叉树中和为某一值的路径(一)
描述
给定一个二叉树root和一个值 sum ,判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点,不能从子节点到父节点
4.总节点数目为n
例如:
给出如下的二叉树, sum=22 sum=22,
返回true,因为存在一条路径 5→4→11→25→4→11→2的节点值之和为 22数据范围:
1.树上的节点数满足 0≤n≤100000≤n≤10000
2.每 个节点的值都满足 ∣val∣≤1000∣val∣≤1000
要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)
进阶:空间复杂度 O(树的高度)O(树的高度),时间复杂度 O(n)O(n)
【解法一】dfs
class Solution { public: bool flag = false; void dfs(TreeNode* root, int sum) { if(root==nullptr)return; sum-=root->val; if(sum==0 && root->left==nullptr && root->right==nullptr) { flag = true; return; } dfs(root->left, sum); dfs(root->right, sum); } bool hasPathSum(TreeNode* root, int sum) { dfs(root, sum); return flag; } };
二叉树中和为某一值的路径(二)
描述
输入一颗二叉树的根节点root和一个整数expectNumber,找出二叉树中结点值的和为expectNumber的所有路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点,不能从子节点到父节点
4.总节点数目为n
示例1
输入:{10,5,12,4,7},22
返回值:[[10,5,7],[10,12]]说明:返回[[10,12],[10,5,7]]也是对的
class Solution {
public:
vector<vector<int>>res;
vector<int> temp;
void dfs(TreeNode* root, int sum)
{
if(root==nullptr)return;
sum-=root->val;
temp.push_back(root->val);
if(sum==0 && root->left==nullptr && root->right==nullptr)
{
res.push_back(temp);
}
dfs(root->left, sum);
dfs(root->right, sum);
sum+=root->val; // 这里进行回溯还原
temp.pop_back();
}
vector<vector<int>> FindPath(TreeNode* root,int expectNumber) {
if(root == nullptr)return res;
dfs(root, expectNumber);
return res;
}
};
二叉树中和为某一值的路径(三)
描述
给定一个二叉树root和一个整数值 sum ,求该树有多少路径的的节点值之和等于 sum 。
1.该题路径定义不需要从根节点开始,也不需要在叶子节点结束,但是一定是从父亲节点往下到孩子节点
2.总节点数目为n
3.保证最后返回的路径个数在整形范围内(即路径个数小于231-1)
【解法一】俩层dfs嵌套
class Solution {
public:
int res = 0;
void dfs(TreeNode* root, int sum)
{
if(root == nullptr)
return;
sum-=root->val;
if(sum==0) // 注意这里,后续可能还会有中和的现象,所以不能直接剪枝
res++;
dfs(root->left, sum);
dfs(root->right, sum);
sum+=root->val;
}
int FindPath(TreeNode* root, int sum) {
// write code here
if(root==nullptr)return 0;
dfs(root, sum);
FindPath(root->left, sum);
FindPath(root->right, sum);
return res;
}
};