- 路径总和 III
给定一个二叉树,它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数。
示例:
root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8
10
/ \
5 -3
/ \
3 2 11
/ \
3 -2 1
返回 3。和等于 8 的路径有:
- 5 -> 3
- 5 -> 2 -> 1
- -3 -> 11
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码
下面是同一种方法,用了两种函数的接口来实现,一种是带着一个数遍历,另外一个是通过计数有返回值。
int pathSum(TreeNode* root, int sum) {
if(root==NULL)return 0;
int count=0;
helper(root,sum,count);
int lres,rres;
lres=pathSum(root->left,sum);
rres=pathSum(root->right,sum);
return count+lres+rres;
}
void helper(TreeNode* T,int sum,int& count){
if(T==NULL)return;
int res=sum-T->val;
if(res==0)count++;
helper(T->left,res,count);
helper(T->right,res,count);
}
int helper2(TreeNode* T,int sum){
if(T==NULL)return 0;
int count,lsum,rsum;
count=0;
if(T->val==sum)count++;
lsum=helper2(T->left,sum-T->val);
rsum=helper2(T->right,sum-T->val);
count=count+lsum+rsum;
return count;
}
int pathSum(TreeNode* root, int sum) {
if(root==NULL)return 0;
int count=helper2(root,sum);
int lres,rres;
lres=pathSum(root->left,sum);
rres=pathSum(root->right,sum);
return count+lres+rres;
}
leetcode112. 路径总和
之前我写的其他解法:
/*递归的解法*/
bool hasPathSum(TreeNode* root, int sum) {
if(root==NULL) return false;
if(root->left==NULL&&root->right==NULL)
return sum==root->val;
else
return (hasPathSum(root->left,sum-root->val)||hasPathSum(root->right,sum-root->val));
}
```
```
/*这个题目也可以用来训练如何使用栈来遍历树*/
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
stack<pair<TreeNode*,int>> s;
TreeNode* temp=root;
while(!s.empty()||temp!=NULL){
while(temp!=NULL){
s.push(make_pair(temp,sum-temp->val));
sum=sum-temp->val;
if(sum==0&&temp->left==NULL&&temp->right==NULL) return true;//除了判断和,还需要判断是否是叶节点,用来针对【1 2】,1的这种情况
temp=temp->left;
}
auto item=s.top();
s.pop();
temp=item.first;
sum=item.second;
temp=temp->right;
}
return false;
}
};
栈解法的修改
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
if(!root)return false;
stack<pair<TreeNode*,int>> s;
s.push(make_pair(root,sum));
TreeNode* T;
int num;
while(!s.empty()){
auto iter=s.top();
s.pop();
T=iter.first;
num=iter.second;
if(T->left==NULL&&T->right==NULL){
if(T->val==num)return true;
}
else{
if(T->left!=NULL)s.push(make_pair(T->left,num-T->val));
if(T->right!=NULL)s.push(make_pair(T->right,num-T->val));
}
}
return false;
}
};