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

最新推荐文章于 2022-03-09 10:18:10 发布

喜欢数理逻辑的医学生Zach

最新推荐文章于 2022-03-09 10:18:10 发布

阅读量216

点赞数

分类专栏： leetcode 文章标签：算法二叉树数据结构 leetcode

本文链接：https://blog.csdn.net/weixin_40883496/article/details/109739696

版权

leetcode 专栏收录该内容

79 篇文章 0 订阅

订阅专栏

1 题目描述

给定一个二叉树和一个值 sum，请找出所有根节点到叶子节点的节点值之和为 sum的路径。

3 代码实现

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

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > pathSum(TreeNode* root, int sum) {
        vector<vector<int>> path_list;
        vector<int> path;
        pathSumHelper(root, sum, path_list, path);
        return path_list;
    }
    void pathSumHelper(TreeNode* root, int sum, vector<vector<int>>& path_list, vector<int> path){
        if(root == NULL)
            return;
        
        path.push_back(root->val);
        
        if(root->left == NULL && root->right == NULL && sum - root->val == 0)
            path_list.push_back(path);
        pathSumHelper(root->left, sum - root->val, path_list, path);
        pathSumHelper(root->right, sum - root->val, path_list, path);
    }
};