376 二叉树的路径和
原题网址:https://www.lintcode.com/problem/binary-tree-path-sum/description
描述
给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值
的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
您在真实的面试中是否遇到过这个题?
样例
给定一个二叉树,和 目标值 = 5
:
1
/ \
2 4
/ \
2 3
返回:
[
[1, 2, 2],
[1, 4]
]
标签
二叉树遍历
二叉树
思路:二叉树遍历节点,递归。
可定义一个函数,递归遍历所有路径(类似于前序遍历,根左右的顺序)。遍历每条路径时遇到叶子结点就判断sum是否为target,是将路径节点值数组压入结果中,继续遍历其他路径;不是,依旧继续遍历其他路径。注意遍历其他路径之前应消除当前步骤的影响,即sum减去当前值(如果定义的函数sum参数不是引用,此步可忽略,见第二段代码),路径节点数组pop掉当前节点值。
AC代码:
/**
* 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>> binaryTreePathSum(TreeNode * root, int target) {
// write your code here
vector<vector<int>> result;
if (root==NULL)
{
return result;
}
vector<int> tmp;
int sum=0;
trav(result,tmp,root,target,sum);
return result;
}
void trav(vector<vector<int>> &result,vector<int> &tmp, TreeNode * root, int target, int & sum)
{
tmp.push_back(root->val);
sum+=root->val;
if (root->left==NULL&&root->right==NULL)
{
if (sum==target)
{
result.push_back(tmp);
}
return ;
}
if (root->left!=NULL)
{
trav(result,tmp,root->left,target,sum);
sum=sum-root->left->val;
tmp.pop_back();
}
if (root->right!=NULL)
{
trav(result,tmp,root->right,target,sum);
sum=sum-root->right->val;
tmp.pop_back();
}
}
};
最开始提交的代码在递归语句之前没有判断root->left、root->right是否为NULL导致只通过27%的数据,提示错误 segmentation fault (core dumped)。想了下应该是如果有左右孩子其中一个为空的话,递归时取val值出错。
2.定义的trav函数sum参数不是引用,AC代码:
/**
* 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>> binaryTreePathSum(TreeNode * root, int target) {
// write your code here
vector<vector<int>> result;
if (root==NULL)
{
return result;
}
vector<int> tmp;
int sum=0;
trav(result,tmp,root,target,sum);
return result;
}
void trav(vector<vector<int>> &result,vector<int> &tmp, TreeNode * root, int target, int sum)
{
tmp.push_back(root->val);
sum+=root->val;
if (root->left==NULL&&root->right==NULL)
{
if (sum==target)
{
result.push_back(tmp);
}
return ;
}
if (root->left!=NULL)
{
trav(result,tmp,root->left,target,sum);
//sum=sum-root->left->val;
tmp.pop_back();
}
if (root->right!=NULL)
{
trav(result,tmp,root->right,target,sum);
//sum=sum-root->right->val;
tmp.pop_back();
}
}
};
其他思路: