力扣 669. 修剪二叉搜索树

最新推荐文章于 2024-08-23 09:59:49 发布

江湖人称贺行风

最新推荐文章于 2024-08-23 09:59:49 发布

阅读量194

点赞数

分类专栏：开始C++吧文章标签： leetcode 算法 c++ 二叉树数据结构

本文链接：https://blog.csdn.net/u010950122/article/details/131505366

版权

开始C++吧专栏收录该内容

185 篇文章 1 订阅

订阅专栏

文章提供了三种C++解决方案来修剪二叉搜索树，去除值不在给定范围(low,high)内的节点。方法包括递归法（两种不同实现）和迭代法。递归法通过对比节点值与边界值来决定修剪方向，而迭代法则通过移动根节点并处理左右子树来达到目标。

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

题目来源：https://leetcode.cn/problems/trim-a-binary-search-tree/description/

C++题解1：递归法。当前节点为空时返回空，不为空时对其值进行分类讨论。以low为例，当前节点值等于low时，意味着其左子树都要丢弃，可指向空；大于low时，说明其左子树也可能满足条件，因此对其左子树进一步递归；小于low时，说明当前节点及其左子树都不满足条件，将当前节点更新为其右子节点。

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

C++题解2：递归法。大致思路同上，较为精简，来源代码随想录。

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

C++题解3：迭代法。见注释，来源代码随想录。

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int L, int R) {
        if (!root) return nullptr;

        // 处理头结点，让root移动到[L, R] 范围内，注意是左闭右闭
        while (root != nullptr && (root->val < L || root->val > R)) {
            if (root->val < L) root = root->right; // 小于L往右走
            else root = root->left; // 大于R往左走
        }
        TreeNode *cur = root;
        // 此时root已经在[L, R] 范围内，处理左孩子元素小于L的情况
        while (cur != nullptr) {
            while (cur->left && cur->left->val < L) {
                cur->left = cur->left->right;
            }
            cur = cur->left;
        }
        cur = root;

        // 此时root已经在[L, R] 范围内，处理右孩子大于R的情况
        while (cur != nullptr) {
            while (cur->right && cur->right->val > R) {
                cur->right = cur->right->left;
            }
            cur = cur->right;
        }
        return root;
    }
};