剑指offer——36二叉树和为某一值的路径

题目描述

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

 

题解：

　　一般涉及到树的题目，首先是想到DFS和BFS了

　　

class Solution {
public:
    vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
        vector<vector<int>>res;
        vector<int>v;
        DFS(root, expectNumber, 0, v, res);
        sort(res.begin(), res.end(), [](vector<int>a, vector<int>b) {
            if (a.size() != b.size())return a.size() > b.size();
            for (int i = 0; i < a.size(); ++i)
                if (a[i] != b[i])
                    return a[i] > b[i];
            return a[0] > b[0];
        });
        return res;        
    }
    void DFS(TreeNode* root, const int N, int sum, vector<int>&v, vector<vector<int>>&res)
    {
        if (root == nullptr)return;
        v.push_back(root->val);//切记，这样算值的递归，一般是先计算值，然后在递归返回时，再将数据减去
        sum += root->val;
        if (sum >= N)
        {
            if (sum == N && root->left == nullptr && root->right == nullptr)//判断是不是叶子节点
                res.push_back(v);
            //return;//不知为何这里不要剪枝，剪枝后结果还是错的
        }
        DFS(root->left, N, sum, v, res);
        DFS(root->right, N, sum, v, res);
        sum -= root->val;
        v.pop_back();
    }
};