LeetCode 113. Path Sum II

Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.

For example:
Given the below binary tree and  sum = 22 ,
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

return

[
   [5,4,11,2],
   [5,8,4,5]
]

题目要求:

给出一个数sum,要求找到所有的满足根到叶子结点的和为sum的路径。

采用深度遍历的方法,需要注意的有以下两点:

1.结点的值可能为负数;

2由于采用的是同一个vector类型的变量temp来存储路径,因此,需要记得出栈。

算法过程如下:

1.判断结点是否为空,如果为空,则返回;

2.将目标值sum减去结点值root->val,并将该结点加入到vector<int>类型的变量temp中。判断,如果sum值为0,且该结点为叶子结点,则将temp放入结果result中。

什么时候出栈呢?这是重点考虑的问题,如果在遍历之前出栈,那么毫无疑问,子结点的路径中不会有该结点,如果在到达子结点的时候出栈,那么将只有叶子结点出栈,向例子中[5,8,4,2]就会加入11这个结点。因此正确的做法应当是,在遍历子树之后出栈。保证,每次进栈都会有一次出栈。

代码实现如下:

void getResult(TreeNode* root,int sum,vector<int> &temp,vector<vector<int> > &result){
	if(root==NULL)
		return ;
	sum=sum - root->val;
	temp.push_back(root->val);
	if(sum==0&&!root->left && !root->right)
		result.push_back(temp);

	getResult(root->left,sum,temp,result);
	getResult(root->right,sum,temp,result);
	temp.pop_back();
}

class Solution {
public:
    vector<vector<int> > pathSum(TreeNode* root, int sum) {
    	vector<vector<int> > result;
    	vector<int> temp;
        getResult(root,sum,temp,result);
        return result;
    }
};



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值