二叉树的路径和
问题描述:
给定一个二叉树,找出所有路径中各节点相加总和等于给定目标值 的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
样例
给定一个二叉树,和 目标值 = 5:
1
/ \
2 4
/ \
2 3
返回:
[
[1, 2, 2],
[1, 4]
]
解题思路:
根据前序遍历的思路,利用局部变量的性质,将某条路径上的从根结点到叶子结点的所有结点的值插入到向量中,迭代此向量,若加和等于所给的目标值的值,则把此向量插到另外一个双重向量中,遍历所有的路径,返回双重向量即可。
代码实现:
class Solution {
public:
/**
* @param root the root of binary tree
* @param target an integer
* @return all valid paths
*/
vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
// Write your code here
vector<vector<int> >paths;
vector<int>path;
vector<int>::iterator it;
if(root==NULL){
return paths;
}
else {
def(root,paths,path,target);
return paths;
}
}
void def(TreeNode*root,vector<vector<int> >&ps,vector<int>p,int target){
p.push_back(root->val);
vector<int>::iterator itp;
if(root->left==NULL&&root->right==NULL){
int flag=0;
for(itp=p.begin();itp!=p.end();itp++){
flag+=*itp;
}
if(flag==target){
ps.push_back(p);
}
}
if(root->left!=NULL){
def(root->left,ps,p,target);
}
if(root->right!=NULL){
def(root->right,ps,p,target);
}
}
};
A题感悟:
在一次加深了对局部变量和全局变量的理解,我觉得这道题比较简单,思路很容易想到,只要注意局部变量和全局变量的区别即可。