题解 | #二叉树根节点到叶子节点和为指定值的路径#

二叉树根节点到叶子节点和为指定值的路径

http://www.nowcoder.com/practice/840dd2dc4fbd4b2199cd48f2dadf930a

dfs。前序遍历,当达到叶子节点并且和满足要求时,保存此路径,递归的过程中还要对路径进行回溯。

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型vector<vector<>>
     */
    void pre_order(TreeNode* root,vector<vector<int>>& result,vector<int>& path,int sum){
        path.push_back(root->val);
        if(root->left==nullptr && root->right==nullptr){
            int a_sum=0;
            for(auto x: path) a_sum+=x;
            if(a_sum==sum) result.push_back(path);
            return;
        }
        if(root->left){
            pre_order(root->left, result, path, sum);
            path.pop_back();
        }
        if(root->right){
            pre_order(root->right, result, path, sum);
            path.pop_back();
        }
    }
    
    vector<vector<int> > pathSum(TreeNode* root, int sum) {
        if(root==nullptr) return {};
        vector<vector<int>> result;
        vector<int> path;
        pre_order(root,result,path,sum);
        return result;
    }
};



全部评论

相关推荐

见见123:简历没有啥问题,是这个社会有问题。因为你刚毕业,没有工作经历,现在企业都不要没有工作经历的。社会病了。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务