时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 32M,其他语言64M 热度指数:579355
本题知识点: 树
算法知识视频讲解
题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解题思路
当前序遍历访问到某一节点时,把该节点添加到路径上,并累加该节点的值。如果该节点为叶节点,且路径中节点的值的和刚好等于输入的整数,则当前路径符合要求,将其加入到最后的结果中。如果当前结点不是叶节点,则继续访问它的子节点。当前结点访问结束后,递归函数将自动回退到它的父节点。因此,在函数退出之前在路径上删除当前节点并减去当前父节点的值,以确保返回父节点时路径刚好是从根节点到父节点。
代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root == NULL || expectNumber == 0)
{
return result;
}
path.push_back(root->val);
expectNumber -= root->val;
如果是叶结点,并且路径上结点的和等于输入的值
// 将这条路径保存到结果result中
if(expectNumber == 0 && root->left == NULL && root->right == NULL)
{
result.push_back(path);
}
// 如果不是叶结点,则遍历它的子结点
FindPath(root->left, expectNumber);
FindPath(root->right, expectNumber);
// 在返回到父结点之前,在路径上删除当前结点,
// 并在currentSum中减去当前结点的值
expectNumber += root->val;
path.pop_back();
return result;
}
};