码迷,mamicode.com
首页 > 其他好文 > 详细

113.Path Sum II

时间:2015-12-13 18:47:52      阅读:144      评论:0      收藏:0      [点我收藏+]

标签:

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]
]
思路:DFS. 如果root为空,直接返回空数组,否则调用help函数。help函数将传入的root节点值压入cur数组,更新当前数组cur的和值now,如果root为叶子节点,且cur数组和为target,则将cur数组压入ret。如果左子结点不为空,递归调用 help(root->left,cur,now,target),右子结点不为空,递归调用 help(root->right,cur,now,target)。请注意,cur数组不能传入引用参数,应该传入形参。每个递归函数在运行过程中对cur数组的操作都是各自为政,互不干涉的。
 
  1. /**
  2. * Definition for a binary tree node.
  3. * struct TreeNode {
  4. * int val;
  5. * TreeNode *left;
  6. * TreeNode *right;
  7. * TreeNode(int x) : val(x), left(NULL), right(NULL) {}
  8. * };
  9. */
  10. class Solution {
  11. public:
  12. vector<vector<int>> ret;
  13. void help(TreeNode *root,vector<int> cur,int now,int target){
  14. now+=root->val;
  15. cur.push_back(root->val);
  16. if(!root->left&&!root->right){
  17. if(now==target)
  18. ret.push_back(cur);
  19. return;
  20. }
  21. if(root->left)
  22. help(root->left,cur,now,target);
  23. if(root->right)
  24. help(root->right,cur,now,target);
  25. }
  26. vector<vector<int>> pathSum(TreeNode* root, int sum) {
  27. if(!root)
  28. return ret;
  29. vector<int> temp;
  30. help(root,temp,0,sum);
  31. return ret;
  32. }
  33. };





113.Path Sum II

标签:

原文地址:http://www.cnblogs.com/zhoudayang/p/5043195.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!