剑指Offer——二叉树中和为某一值的路径

题目描述

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

思路：先序遍历，边界条件是叶子节点、空节点、路径值吻合节点。左右子树遍历完要回退。

class Solution {
private:
    int count=0;
    vector<vector<int> > result;
    vector<int> temp;
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        if(!root)
            return result;
        count+=root->val;
        temp.push_back(root->val);
        if(!root->left&&!root->right&&count==expectNumber)
        {//叶子节点，并且和期待结果一致
            result.push_back(temp);
        }
        FindPath(root->left, expectNumber);
        FindPath(root->right, expectNumber);
        //左右子树遍历完了，count和temp都要回退
        count-=root->val;
        temp.pop_back();
        return result;
    }
};