LeetCode 112. 113. 路经总和

最新推荐文章于 2024-07-30 09:12:21 发布

_almost__

最新推荐文章于 2024-07-30 09:12:21 发布

阅读量63

点赞数

分类专栏：刷题

本文链接：https://blog.csdn.net/almost_afei/article/details/118459437

版权

刷题专栏收录该内容

293 篇文章 0 订阅

订阅专栏

本文介绍了两个关于二叉树路径总和的问题，分别是路径总和和路径总和II。首先，针对路径总和的错误解法进行了分析，指出其仅检查当前节点为空而未验证叶子节点的条件。接着，提供了正确的解决方案，当遇到叶子节点且路径和等于目标值时返回true。对于路径总和II，文章给出了正确解法，使用深度优先搜索和回溯方法找出所有路径，并存储在结果向量中。

摘要由CSDN通过智能技术生成

文章目录

- 112. 路径总和
- 113. 路径总和 II

112. 路径总和

难度：简单。
标签：深度优先搜索，二叉树。

先放一个错误的解法：

class Solution {

    bool dfs(TreeNode* root, int k){
        if(root == nullptr && k == 0){
            return true;
        }
        if(root == nullptr)return false;

        return dfs(root->left, k - root->val) || dfs(root->right, k - root->val);
    }

public:
    bool hasPathSum(TreeNode* root, int targetSum) {
        return dfs(root, targetSum);
    }
};

对于以下用例，结果返回的是true

[5,4,8,11,null,13,4,7,2,null,null,null,1]
9

true的路径是
在这里插入图片描述
因为上述代码只是判断了当前的节点为空且k为0，就返回true，而当前节点为空并不能证明其上一个点就是叶子结点。证明叶子结点还是需要判断节点的左右子树是否都为空。

正确解法：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {

    bool dfs(TreeNode* root, int k){
        if(root == nullptr)return false;
        if(root->left == nullptr && root->right == nullptr && k == root->val){
            return true;
        }
        return dfs(root->left, k - root->val) || dfs(root->right, k - root->val);
    }

public:
    bool hasPathSum(TreeNode* root, int targetSum) {
        return dfs(root, targetSum);
    }
};

结果：
在这里插入图片描述

113. 路径总和 II

难度：中等。
标签：二叉树，深度优先搜索，回溯。

正确解法：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
    vector<vector<int>> ans;

    void dfs(TreeNode* node, vector<int>& path, int k){
        if(node == nullptr)return;

        path.emplace_back(node->val);
        if(node->left == nullptr && node->right == nullptr && k == node->val){
            ans.emplace_back(path);
        }
        dfs(node->left, path, k - node->val);
        dfs(node->right, path, k - node->val);
        path.pop_back();
    }

public:
    vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
        vector<int> path;
        dfs(root, path, targetSum);
        return ans;
    }
};