描述:
给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值 的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
样例:给定一个二叉树,和 目标值 = 5:
1
/ \
2 4
/ \
2 3
返回:
[
[1, 2, 2],
[1, 4]
]
实现思路:
就像之前写的所有路径,我们需要找到符合条件的路径,就是需要把每条路径的和算出来,如果等于指定值我们就把他加入到va向量里面。但是,每次不要忘了删除上一条路径。
实现代码:
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param root the root of binary tree
* @param target an integer
* @return all valid paths
*/
vector<vector<int>> va;
void pathsum(TreeNode *root,int sum,int goal,vector<int> v)
{
if(root->left==NULL && root->right==NULL)
{
if(sum==goal)
va.push_back(v);
return;
}
int s;
if(root->left!=NULL)
{
vector<int> vm(v);
s=sum+root->left->val;
vm.push_back(root->left->val);
pathsum(root->left,s,goal,vm);
}
if(root->right!=NULL)
{
vector<int> vm(v);
s=sum+root->right->val;
vm.push_back(root->right->val);
pathsum(root->right,s,goal,vm);
}
}
vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
// Write your code here
if(root==NULL)
return va;
vector<int>v;
v.push_back(root->val);
pathsum(root,root->val,target,v);
return va;
}
};
做题感想:通过这个题,进一步体会vector向量的用法。进一步体会pop_back的用法。