标签: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
,
5 / 4 8 / / 11 13 4 / \ 7 2 1
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
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); } };
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] ]
解题分析:
现在需要求出满足条件的路径,并且是所有路径,所以左子树求到了满意结果,不能return,要接着求右子树
我们在上一题中,最后叶子结点如果满足 val == sum 那么这条路径就符合要求,此时我们将结果添加到result中
但是如果保存该路径上的所有结点值呢? 我们是 每遇到一个非空结点都添加到一个vector中(其实这个vector就是一个栈)
如果栈顶(vector最末元素)的左右子树都不存在满足条件的路径,那么就出栈,也就是 从其父节点开始继续判断
这就是类似于 二叉树的先序遍历,每遇到一个结点就入栈,随后如果不满足条件就出栈达到其父节点
先序遍历对应 根左右,也就是说最后的结果就是 从根节点出发的路径,即路径方向也吻合,而不需要逆向
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(); } };
Leetcode:Path Sum 二叉树路径和,布布扣,bubuko.com
标签:style class blog code java http
原文地址:http://www.cnblogs.com/wwwjieo0/p/3784690.html