leetcode之二叉搜索树的范围和（C++）

最新推荐文章于 2024-10-01 05:04:32 发布

cqu_shuai

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量104

点赞数

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

本文链接：https://blog.csdn.net/beilizhang/article/details/116200271

版权

leetcode重点250题专栏收录该内容

111 篇文章 2 订阅

订阅专栏

参考链接

https://leetcode-cn.com/problems/range-sum-of-bst/
https://leetcode-cn.com/problems/range-sum-of-bst/solution/er-cha-sou-suo-shu-de-fan-wei-he-by-leet-rpq7/

题目描述

给定二叉搜索树的根结点 root，返回值位于范围 [low, high] 之间的所有结点的值的和。
在这里插入图片描述

解题思路

中序遍历

关于二叉搜索树的问题，第一反应就是中序遍历，恰好可以解决本题。由于中序遍历会得到递增的序列，所以只需要在遍历过程中加上判断，如果值在区间内则加上它，如果大于high，则直接退出，后续必定没有区间内的值了。

深度优先搜索

其实就是递归的思路，如果当前节点的值在区间内，那返回节点的值再加上左右子树的范围和；如果当前节点的值大于right，右子树的范围和必定为0，返回左子树的范围和；如果当前节点的值小于left，左子树的范围和必定为0，返回右子树的范围和。

代码

中序遍历

class Solution {
public:
    int sum = 0;
    void traverse(TreeNode* root, int low, int high)
    {
        if (root == nullptr)
        {
            return;
        }
        traverse(root->left, low, high);
        if (root->val >= low && root->val <= high)
        {
            sum += root->val;
        }
        if (root->val > high)
        {
            return;
        }
        traverse(root->right, low, high);
    }
    int rangeSumBST(TreeNode* root, int low, int high) {
        traverse(root, low, high);
        return sum;
    }
};

深度优先搜索

class Solution {
public:
    int rangeSumBST(TreeNode *root, int low, int high) {
        if (root == nullptr) {
            return 0;
        }
        if (root->val > high) {
            return rangeSumBST(root->left, low, high);
        }
        if (root->val < low) {
            return rangeSumBST(root->right, low, high);
        }
        return root->val + rangeSumBST(root->left, low, high) + rangeSumBST(root->right, low, high);
    }
};