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

一、题目

输入一棵二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。

从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

保证树中结点值均不小于 0。

数据范围

树中结点的数量 [0,1000]。

样例

给出二叉树如下所示，并给出num=22。
      5
     / \
    4   6
   /   / \
  12  13  6
 /  \    / \
9    1  5   1

输出：[[5,4,12,1],[5,6,6,5]]

二、思路

 遍历二叉树，满足条件的path路径放到ans数组中。

满足条件：叶子结点并且sum为题目要求。

三、代码实现

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    //定义数组存储答案 ans
    vector<vector<int>> ans;
    //定义数组存储路径 path 
    vector<int> path;
    
    vector<vector<int>> findPath(TreeNode* root, int sum) {
        //从根节点开始遍历 最后返回答案数组ans
        dfs(root, sum);
        return ans;
    }
    
    void dfs(TreeNode *root,int sum)
    {
        //递归不进行条件 当前root为空
        if(!root) return;
        //将当前点的值放到路径数组
        path.push_back(root -> val);
        //sum 减去当前遍历到的值
        sum -= root -> val;
        //当是叶子结点并且sum的值为0 说明满足条件 在ans数组添加这个路径path
        if(!root -> left && !root -> right && !sum) ans.push_back(path);
        //继续遍历左子树 右子树
        dfs(root -> left, sum);
        dfs(root -> right, sum);
        //还原path 便于下次遍历
        path.pop_back();
    }
};