问题描述:
给定一个二叉树,找出所有路径中各节点相加总和等于给定目标值的路径。
一个有效的路径,指的是从根结点到叶节点的路径。
实现思路:
找出所有路径,算出路径和,如果路径和等于目标值那么就将该路径加入vector中。最后返回该vector。
/**
* 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>>P;
vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
// Write your code here
vector<int>v;
int s=0;
find(root,target,v,s);
return P;
}
void find(TreeNode *root,int target,vector<int>v,int s)
{
if(root!=NULL)
{
v.push_back(root->val);
s=s+root->val;
if(s==target&&root->left==NULL&&root->right==NULL)
{
P.push_back(v);
}
if(root->left!=NULL)
{
find(root->left,target,v,s);
}
if(root->right!=NULL)
{
find(root->right,target,v,s);
}
s=s-root->val;
v.pop_back();
}
}
};
感想:
要注意清理现场,包括路径和与路径。