Leetcode---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]
]



Show Tags
Have you met this question in a real interview?

这道题关键的地方在于如何传递一个vector,让它在递归中不停的存数据,递归完了之后,又能弹出之前的数据。

所以,一个关键的技巧就是不传引用,而是传值。这样,每次递归的时候,实际上都会在栈上生成一个新的参数,递归结束的时候,原传入参数没有受到任何影响。再回到上一层,就像是弹出了。

好像是在说天书,好吧,看下面两个实现,一个是用引用类型的,一个是用值类型的,都Accept了
引用类型:

  1. vector<vector<int> > result;
  2. void pathSumHelper(TreeNode *root, int sum, vector<int>& tmp){
  3.     if(root==NULL)
  4.         return;
  5.     if(root->left==NULL &&root->right==NULL){
  6.         if(sum==root->val){
  7.             tmp.push_back(root->val);
  8.             result.push_back(tmp);
  9.         }
  10.         return;
  11.     }
  12.     sum-=root->val;
  13.     tmp.push_back(root->val);
  14.     vector<int> tmp2(tmp);
  15.     vector<int> tmp3(tmp);
  16.     pathSumHelper(root->left,sum,tmp2);
  17.     pathSumHelper(root->right,sum,tmp3);
  18.     sum+=root->val;
  19.     tmp.pop_back();

  20. }
  21. vector<vector<int> > pathSum(TreeNode *root, int sum) {
  22.     vector<int> tmp;
  23.     pathSumHelper(root,sum,tmp);
  24.     return result;
  25. }
其实就是在传入的时候copy了一把。但是注意,每个函数都要copy一把,否则又会错。

然后是值类型的:

  1. vector<vector<int> > result;
  2. void pathSumHelper(TreeNode *root, int sum, vector<int> tmp){
  3.     if(root==NULL)
  4.         return;
  5.     if(root->left==NULL &&root->right==NULL){
  6.         if(sum==root->val){
  7.             tmp.push_back(root->val);
  8.             result.push_back(tmp);
  9.         }
  10.         return;
  11.     }
  12.     sum-=root->val;
  13.     tmp.push_back(root->val);
  14.     pathSumHelper(root->left,sum,tmp);
  15.     pathSumHelper(root->right,sum,tmp);
  16.     sum+=root->val;
  17.     tmp.pop_back();

  18. }
  19. vector<vector<int> > pathSum(TreeNode *root, int sum) {
  20.     vector<int> tmp;
  21.     pathSumHelper(root,sum,tmp);
  22.     return result;
  23. }

非常整齐的代码,但是因为是值传递,第17行 tmp . pop_back ( ) ;可以不要,因为函数结束了tmp其实也销毁了。

最近又写了一个

vector<vector<int> > result;
void help(TreeNode *root, int& num, int &sum, vector<int>& temp){
    if(root==NULL){
    }
    else if(root->left==NULL && root->right==NULL){
        num+=root->val;
        temp.push_back(root->val);
        if(num==sum)
            result.push_back(temp);
        temp.pop_back();
        num-=root->val;
    }
    else{
        temp.push_back(root->val);
        num+=root->val;
        help(root->left,num,sum,temp);
        help(root->right,num,sum,temp);
        num-=root->val;
        temp.pop_back();
    }
}
vector<vector<int> > pathSum(TreeNode *root, int sum) {
    vector<int> temp;
    int num=0;
    help(root,num,sum,temp);
    return result;
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值