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

Leetcode:Path Sum 二叉树路径和

时间:2014-06-13 16:09:02      阅读:237      评论:0      收藏:0      [点我收藏+]

标签:style   class   blog   code   java   http   

Path Sum:

Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum.

For example:
Given the below binary tree and sum = 22,

bubuko.com,布布扣
              5
             /             4   8
           /   /           11  13  4
         /  \              7    2      1
bubuko.com,布布扣

 

return true, as there exist a root-to-leaf path 5->4->11->2 which sum is 22.

 

解题分析:

二叉树首先应该想到递归

对于root = 5, 寻找sum=22的路径 这个问题可以分为  root=4 寻找sum=22-5=17的路径 和 root=8 寻找sum=22-5=17的路径 这两部分

也就是说,我们可以递归寻找左右子树路径和,我们只需要更改和值即可

终止条件:如果root为空,立即返回false, 如果root为叶子节点,如果其值为最后剩余的和值,那么就存在总的路径和值等于sum

bubuko.com,布布扣
class Solution {
public:
    bool hasPathSum(TreeNode *root, int sum) {
        if (root == nullptr) return false;
        if (root->left == NULL && root->right == NULL) return sum == root->val;
        return hasPathSum(root->left, sum - root->val) || hasPathSum(root->right, sum - root->val);
    }
};
bubuko.com,布布扣

 

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,

bubuko.com,布布扣
              5
             /             4   8
           /   /           11  13  4
         /  \    /         7    2  5   1
bubuko.com,布布扣

 

return

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

解题分析:

现在需要求出满足条件的路径,并且是所有路径,所以左子树求到了满意结果,不能return,要接着求右子树

我们在上一题中,最后叶子结点如果满足 val == sum 那么这条路径就符合要求,此时我们将结果添加到result中

但是如果保存该路径上的所有结点值呢? 我们是 每遇到一个非空结点都添加到一个vector中(其实这个vector就是一个栈)

如果栈顶(vector最末元素)的左右子树都不存在满足条件的路径,那么就出栈,也就是 从其父节点开始继续判断

这就是类似于 二叉树的先序遍历,每遇到一个结点就入栈,随后如果不满足条件就出栈达到其父节点

先序遍历对应 根左右,也就是说最后的结果就是 从根节点出发的路径,即路径方向也吻合,而不需要逆向

bubuko.com,布布扣
class Solution {
public:
    vector<vector<int> > pathSum(TreeNode *root, int sum) {
        vector<vector<int>> result;
        vector<int> tmp;
        BintreePathSum(root, sum, tmp, result);
        return result;
    }
    
    void BintreePathSum(TreeNode* root, int gap, vector<int>& tmp, vector<vector<int>>& result) {
        if (root == NULL) return;
        tmp.push_back(root->val);
        
        if (root->left == NULL && root->right == NULL) {
            if (root->val == gap) {
                result.push_back(tmp);
            }
        }
        BintreePathSum(root->left, gap - root->val, tmp, result);
        BintreePathSum(root->right, gap - root->val, tmp, result);
        
        tmp.pop_back();
    }
};
bubuko.com,布布扣

 

Leetcode:Path Sum 二叉树路径和,布布扣,bubuko.com

Leetcode:Path Sum 二叉树路径和

标签:style   class   blog   code   java   http   

原文地址:http://www.cnblogs.com/wwwjieo0/p/3784690.html

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