【leetcode c++】113 Path Sum II

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,


Return


Path Sum二号题刚好和一号题连在一起,就顺便做了。这次要求是的把路径记下来。所以呢,路径可能不止一条,所以必须要把整棵树遍历。所谓的路径是一个root-to-leaf path。所以从根节点开始,先保证有一个数组来存放路径。但是我们把这个数组当作栈来用,也就是使用push_back、pop_back。每进入一个节点就push,离开以这个节点为根的树就pop,到达叶子节点就判断是否是所求路径。

这题其实不是很难想,应该考验的是你如何减少无用的时间空间消耗。因为,题目要求的结果要放在vector二维数组中。策略不对可能就要多耗时。

开始


比如这棵树,求sum=6,结果应该是2个[1,2,3]

从根节点遍历到第一个(从左到右)叶子节点时,这个路径是符合的,那么我们需要把这个路径保存下来,也就是复制一个路径出来,把这个副本作为结果保存起来,原来的路径则在遍历回节点2的时候pop操作。

其实也就是,深度优先。这个在数据结构清华大学出版社的书里面的栈的那部分,有一题迷宫求出口的题,也是深度优先来解。这个题其实也可以理解成一个迷宫求路径的问题。只不过,这个‘迷宫‘路径有很多条,判定出口规则也不一样。


**************************


**************************


**************************


**************************


**************************


我们发现,保存路径的那个vector数组时刻在变化,只有找到合法路径的时候,才会丢一个副本进入结果二维数组中,所以这个路径数组我们应当让它唯一,所以我传了引用。找到了合法路径就直接push_back即可,因为push_back进去的本来就只是个副本,我们可以传引用。

代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
    vector<vector<int>> res;
    if(NULL == root) return res;
    vector<int> temp;
    int preSum = 0;
    Lv(root, sum, preSum, temp, res);
    }
    
    void Lv(TreeNode* root,int sum, int& preSum, vector<int>& temp, vector<vector<int>>& res) {
    if(NULL == root) return;
    //进入节点就push
    preSum += root->val;
    temp.push_back(root->val);
    
    if(!root->left && !root->right)
    {
        if(sum == preSum)
        {
            res.push_back(temp);//路径找到,copy到二维数组中
        }
        preSum -= root->val;
        temp.pop_back();
        return;
    }
    
    Lv(root->left, sum, preSum, temp, res);
    Lv(root->right, sum, preSum, temp, res);
    //离开节点就pop
    preSum -= root->val;
    temp.pop_back();
    }
};

2015-07-28


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值