给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值 的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
样例
给定一个二叉树,和 目标值 = 5:
1
/
2 4
/
2 3
返回:
[ [1, 2, 2], [1, 4] ]
思路:采用递归思想。分两部分,一部分寻找二叉树中的路径(先序遍历),一部分判断路径是否满足要求。
/**
* 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>> vec;
vector<int> temp;
void Is_equal(vector<int> v,int ans){
//判断路径是否满足要求;
int sum=0;
for(int i=0;i<v.size();i++){
sum+=v[i];
}
if(sum==ans){
vec.push_back(v);
}
}
vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
// Write your code here
//遍历二叉树,寻找路径并存放入临时容器temp中,并判断是否满足要求;
if(root==NULL)
{
return vec;
}
temp.push_back(root->val);
if(root->left!=NULL){
binaryTreePathSum(root->left,target);
}
if(root->right!=NULL){
binaryTreePathSum(root->right,target);
}
if(root->right==NULL&&root->left==NULL){
Is_equal(temp,target);
}
temp.erase(temp.begin()+temp.size()-1);//清空temp容器,空间和元素都清空;
return vec;
}
};