牛客网-二叉树中和为某一值的路径

题目描述

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

 

 

解题思路

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> path;
    void travel(TreeNode* root,int k){
        if(root == NULL) return;
        if(root->left == NULL && root->right == NULL){
            path.push_back(root->val);
            int sum = accumulate(path.begin(),path.end(),0);
            if(sum == k) ans.push_back(path);
            path.pop_back();
            return;
        }
        //回溯
        path.push_back(root->val);
        travel(root->left,k);
        travel(root->right,k);
        path.pop_back();
    }
    vector<vector<int> > FindPath(TreeNode* root,int k) {
        travel(root,k);
        return ans;
    }
};