Chinaunix首页 | 论坛 | 博客
  • 博客访问: 116755
  • 博文数量: 53
  • 博客积分: 0
  • 博客等级: 民兵
  • 技术积分: 620
  • 用 户 组: 普通用户
  • 注册时间: 2014-08-24 16:22
文章存档

2014年(53)

我的朋友

分类: C/C++

2014-12-13 17:52:36

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其实也销毁了。

阅读(1174) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~