二叉树中和为某一值的路径

题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

class Solution {
public:
static bool compare_size(vector&a,vector&b){
return a.size()>b.size();
}
void dfs(TreeNode*root,vector<vector >&res,vector&tmp,int sum,int count){

tmp.push_back(root->val);
count+=root->val;
if(root->left!=NULL)
dfs(root->left,res,tmp,sum,count);

if(root->right!=NULL)
dfs(root->right,res,tmp,sum,count);

if(root->leftNULL&&root->rightNULL){
if(count==sum){
res.push_back(tmp);
}
}
tmp.pop_back();

}
vector<vector > FindPath(TreeNode* root,int expectNumber) {
vector<vector >res;
vectortmp;
if(root==NULL)
return res;
int count=0;
dfs(root,res,tmp,expectNumber,count);
sort(res.begin(),res.end(),compare_size);
return res;
}
};

posted @ 2019-10-02 14:36  夜深不自知  阅读(105)  评论(0)    收藏  举报