【剑指offer-C++】JZ82:二叉树中和为某一值的路径[一]
- 题目描述
- 解题思路
题目描述
描述:给定一个二叉树root和一个值 sum ,判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点;
2.叶子节点是指没有子节点的节点;
3.路径只能从父节点到子节点,不能从子节点到父节点;
4.总节点数目为n;
例如:给出如下的二叉树, sum=22,
返回true,因为存在一条路径 5→4→11→2的节点值之和为 22。
数据范围:
1.树上的节点数满足 0≤n≤10000;
2.每 个节点的值都满足 ∣val∣≤1000;
要求:空间复杂度 O(n),时间复杂度 O(n)。
进阶:空间复杂度 O(树的高度),时间复杂度 O(n)。
输入:{5,4,8,1,11,#,9,#,#,2,7},22
返回值:true
输入:{1,2},0
返回值:false
输入:{1,2},3
返回值:true
输入:{},0
返回值:false
解题思路
二叉树中和为某一值的路径(一):最直观的想法是,因为要找从根节点到叶子节点的路径,故直接采用前序遍历即可。当当前节点为空节点则直接返回false,反之当当前结点为叶子节点即当前节点不为空但是当前节点的左右子树均为空并且当前节点值等于给定值则表明当前路径满足要求故返回true。接着依次遍历当前左或右子树找目标和为给定值减去当前节点值的路径是否存在并返回即可。
bool dfs(TreeNode* root, int sum)
{
// 一定加上这句 即当前节点为空 直接返回false 因为一定是前面不满足才继续调用的
if(!root)
return false;
// 叶子结点且其值等于给定求和
if(root && !root->left && !root->right && root->val==sum)
return true;
// 如果在这里加上叶子结点且其值不等于给定和返回false只是一部分 没有给定单亲结点的情况
bool left = dfs(root->left,sum-root->val);
bool right = dfs(root->right,sum-root->val);
return left || right;
}
bool hasPathSum(TreeNode* root, int sum)
{
// 空数返回false
if(!root)
return false;
return dfs(root,sum);
}