二叉树根节点到叶子节点和为指定值的路径(先序遍历)

Stdleohao

于 2021-04-13 19:12:08 发布

阅读量597

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/stdleohao/article/details/115675868

nowcode 专栏收录该内容

38 篇文章

订阅专栏

本文介绍如何使用递归的先序遍历方法解决二叉树中找到所有根节点到叶子节点路径，其节点值之和等于给定值sum的问题。通过实例代码展示并指出之前的错误——res数组未用值传递，最终提供正确实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个二叉树和一个值\ sum sum，请找出所有的根节点到叶子节点的节点值之和等于\ sum sum 的路径，
例如：
给出如下的二叉树， sum=22，

返回 [ [5,4,11,2], [5,8,9] ]

思路

利用递归的先序遍历，每次储存遍历过的节点，并用sum去减，最后判断是否为0

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

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型vector<vector<>>
     */
    // 先序遍历，
    void dfs(TreeNode* root, int sum, vector<vector<int>> &res, vector<int> path){
        if(root == NULL)
            return ;
        path.push_back(root->val);
        if(!root->left && !root->right && sum == root->val)
            res.push_back(path);
        dfs(root->left, sum-root->val, res, path);
        dfs(root->right, sum-root->val, res, path);
        
    }
    
    vector<vector<int> > pathSum(TreeNode* root, int sum) {
        // write code here
        vector<vector<int>> res;
        vector<int> path;
        dfs(root, sum, res, path);
        return res;
    }
};