二叉树中和为某一值的路径
1.题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
2.思路
利用二叉树的深度优先遍历,累加当前的和currSum,如果到达叶子节点并且当前和currSum等于expectSum,则把路径插入结果数组vec中,否则回溯,并把currSum减去当前节点的值。
3.代码
/*
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> > FindPath(TreeNode* root,int expectNumber) {
vector<vector<int>> res;
if(root == NULL){
return res;
}
vector<int> path;
int currSum = 0;
_FindPath(root,res,path,expectNumber,currSum);
return res;
}
void _FindPath(TreeNode* root,vector<vector<int>>& res,vector<int> path,int& expectNumber,int& currSum){
if(root == NULL){
return;
}
path.push_back(root->val);
currSum += root->val;
bool isLeaf = (root->left == NULL) && (root->right == NULL);
if(isLeaf && (currSum == expectNumber)){
res.push_back(path);
}
_FindPath(root->left,res,path,expectNumber,currSum);
_FindPath(root->right,res,path,expectNumber,currSum);
currSum -= root->val;
}
};
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(n)