问题描述:
给定一个二叉树,找出所有路径中各节点相加总和等于给定
目标值
的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
样例:
给定一个二叉树,和 目标值 = 5
:
1
/ \
2 4
/ \
2 3
返回:
[
[1, 2, 2],
[1, 4]
]
实现思路:
从根节点开始遍历二叉树,并将路径存放到向量v中,当路径和sum等于给定目标值并且当前节点的左右子树都为空,也就是当前节点就是叶子节点了,此时将向量v存放到向量V中。若当前节点的左右子树不全为空,继续调用自身,直至是叶子节点。
实现代码:
/**
* 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>> V;
vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
// Write your code here
vector<int> v;
if(root==NULL)
return V;
int sum=0;
find(root,target,v,sum);
return V;
}
void find(TreeNode *root,int target,vector<int> v,int sum)
{
v.push_back(root->val);
sum+=root->val;
if(sum==target&&root->left==NULL&&root->right==NULL)
{
V.push_back(v);
}
if(root->left!=NULL)
{
find(root->left,target,v,sum);
}
if(root->right!=NULL)
{
find(root->right,target,v,sum);
}
}
};
* 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>> V;
vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
// Write your code here
vector<int> v;
if(root==NULL)
return V;
int sum=0;
find(root,target,v,sum);
return V;
}
void find(TreeNode *root,int target,vector<int> v,int sum)
{
v.push_back(root->val);
sum+=root->val;
if(sum==target&&root->left==NULL&&root->right==NULL)
{
V.push_back(v);
}
if(root->left!=NULL)
{
find(root->left,target,v,sum);
}
if(root->right!=NULL)
{
find(root->right,target,v,sum);
}
}
};
做题感想:
这个题也是返回以向量为元素的向量,但是用搞清楚全局变量和局部变量。对这个题来说,二维向量是全局变量,因为它要返回函数内每一条路径和等于给定目标值的路径。