Leetcode112. 路径总和

最新推荐文章于 2024-08-24 13:50:23 发布

原创最新推荐文章于 2024-08-24 13:50:23 发布 · 1.4k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #递归 #广度优先搜索

Every day a LeetCode 专栏收录该内容

550 篇文章

订阅专栏

本文介绍了112.路径总和问题的两种解决方案：递归方法和广度优先搜索(BFS)方法。递归方法通过逐步减少目标和并检查叶子节点来解决；BFS则利用队列跟踪节点及其路径和，直到找到符合条件的路径。

Every day a leetcode

题目来源：112. 路径总和

解法1：递归

hasPathSum 函数：询问是否存在从当前节点 root 到叶子节点的路径，满足其路径和为 sum。

假定从根节点到当前节点的值之和为 val，我们可以将这个大问题转化为一个小问题：是否存在从当前节点的子节点到叶子的路径，满足其路径和为 targetSum - val。

于是，这个问题可以用递归实现。

退出条件：走到了空结点

当遇到叶子节点时，判断该值是否等于剩余的targetSum。是，返回 true，不是返回 false。

递归判断节点的左子树和右子树，剩余的targetSum=targetSum-root->val。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */


bool hasPathSum(struct TreeNode* root, int targetSum){
    if(root == NULL) return false;
    if(root->left == NULL && root->right == NULL) return root->val == targetSum;
    return hasPathSum(root->left,targetSum-root->val) || hasPathSum(root->right,targetSum-root->val);
}

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

解法2：BFS

使用广度优先搜索的方式，记录从根节点到当前节点的路径和.

使用两个队列，分别存储将要遍历的节点，以及根节点到这些节点的路径和。

代码：

/**
 * 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 {
public:
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root == NULL) return false;
        queue<TreeNode*> node;
        queue<int> value;
        node.push(root);
        value.push(root->val);
        while(!node.empty())
        {
            TreeNode* curNode=node.front();
            int curValue=value.front();
            node.pop();
            value.pop();
            if(curNode->left == NULL && curNode->right == NULL)
            {
                if(curValue == targetSum) return true;
            }
            if(curNode->left)
            {
                node.push(curNode->left);
                value.push(curValue+curNode->left->val);
            }
            if(curNode->right)
            {
                node.push(curNode->right);
                value.push(curValue+curNode->right->val);
            }
        }
        return false;
    }
};