剑指Offer 二叉树中和为某一值的路径(dfs)

题目描述

输入一颗二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

 

 

思路：

递归，然后深搜，因为题目定义的，是从根到叶子节点的才算一条路径，所以，搜索到叶子节点的时候return ，每次把sum-当前节点的值，传给dfs函数。

 

class Solution {
public:
    vector<vector<int> > ans;
    vector<int> path;
    
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {        
        dfs(root,expectNumber);
        return ans;
    }
    
    void dfs(TreeNode* node,int sum)
    {
        if(node==NULL)
            return;
        path.push_back(node->val);
        if(!node->left&&!node->right&&sum-node->val==0)
            ans.push_back(path);
        else
        {
            if(node->left)
                dfs(node->left,sum-node->val);
            if(node->right)
                dfs(node->right,sum-node->val);    
        }
        path.pop_back();
        
    }
};