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

Posted 2020-11-15 qiuhaifeng

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

class Solution {
    public:
    static bool compare_size(vector<int>&a,vector<int>&b){
        return a.size()>b.size();
    }
    void dfs(TreeNode*root,vector<vector<int> >&res,vector<int>&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->left==NULL&&root->right==NULL){
            if(count==sum){
                res.push_back(tmp);
            }
        }
        tmp.pop_back();

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