牛客网刷题|二叉树中和为某一值的路径

题目来源：牛客网

编程连接

题目描述

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

解析：

经典DFS算法，深度优先算法

class Solution {
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        vector<vector<int>>ret;
        vector<int>path;
        dfs(ret,root,path,expectNumber,0,0);
        return ret;
    }

    void dfs(vector<vector<int>>&ret,TreeNode* root,vector<int>&path,int expectNumber,int len,int sum)
    {
        if(root == nullptr)
            return;
        if(len == path.size())
            path.push_back(root->val);
        else
            path[len] = root->val;

        if(sum + root->val == expectNumber && root->left == nullptr && root->right == nullptr)
        {
            vector<int>arr(len+1);
            for(int i=0;i<len+1;i++)
            {
                arr[i] = path[i];
            }
            ret.push_back(arr);
        }

        dfs(ret,root->left,path,expectNumber,len+1,sum + root->val);
        dfs(ret,root->right,path,expectNumber,len+1,sum + root->val);       
    }
};