问题描述:
给定一个二叉树,找出所有路径中各节点相加总和等于给定目标值
的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
样例给定一个二叉树,和 目标值 = 5
:
1
/ \
2 4
/ \
2 3
返回:
[
[1, 2, 2],
[1, 4]
]
解题思路:
建立一个二维向量,用以储存符合要求的路径,路径表示为一维向量的形式。首先判断根节点的左右子树是否为空,将根节点的数值放入一维向量中,如果左右子树都为空判断根节点的数值是否等于给定的数值,若等于给定的数值,则将其放入二维向量中。定义一个整数型t,用以计算二叉树路径的长度。如果左子树不为空,则将左子树节点的数值加在t上,同时将左子树节点放入一维向量中,然后继续判断,继续调用函数,直到叶子节点,即左右子树都为空,然后判断路径长度是否等于给定的数值,若满足条件,将这条路径对应的一维向量放入二维向量中。对于右子树的情况做相同操作。
代码实现:
/**
* 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>>lz;
vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
// Write your code here
if(root==NULL)
return lz;
vector<int>l;
func(root,root->val,target,l);
return lz;
}
void func(TreeNode *root,int t,int target,vector<int> l){
l.push_back(root->val);
if(root->left==NULL&&root->right==NULL){
if(t==target)
lz.push_back(l);
return;
}
int s;
if(root->left!=NULL){
s=t+root->left->val;
func(root->left,s,target,l);
}
if(root->right!=NULL){
s=t+root->right->val;
func(root->right,s,target,l);
}
}
};
解题感悟:
需要在给定的函数中调用一个新的函数,新函数写在给定函数体外,若直接在给定的函数体内写操作代码,会出现restricted call错误。错误情况如下:
if(root==NULL) return lz;
else{
vector<int>l;int t=root->val;
l.push_back(root->val);
if(root->left==NULL&&root->right==NULL&&t==target){
lz.push_back(l);
return lz;
}
if(root->left!=NULL){
t+=root->left->val;
binaryTreePathSum(root->left,target);
}
if(root->right!=NULL){
t+=root->right->val;
binaryTreePathSum(root->right,target);
}
}
}
};