LeetCode 669. 修剪二叉搜索树-CSDN博客

本文链接：https://blog.csdn.net/2301_80211119/article/details/139012728

LeetCode 669. 修剪二叉搜索树

1、题目

题目链接：669. 修剪二叉搜索树
给你二叉搜索树的根节点 root ，同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树，使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即，如果没有被移除，原有的父代子代关系都应当保留)。可以证明，存在 唯一的答案 。
所以结果应当返回修剪好的二叉搜索树的新的根节点。注意，根节点可能会根据给定的边界发生改变。

示例 1：

输入：root = [1,0,2], low = 1, high = 2
输出：[1,null,2]

示例 2：

输入：root = [3,0,4,null,2,null,null,1], low = 1, high = 3
输出：[3,2,null,1]

提示：

树中节点数在范围 [1, 104] 内
0 <= Node.val <= 104
树中每个节点的值都是唯一的
题目数据保证输入是一棵有效的二叉搜索树
0 <= low <= high <= 104

2、深度优先搜索（递归法）

思路

对根结点 root 进行深度优先遍历。
对于当前访问的结点，如果结点为空结点，直接返回空结点；
如果结点的值小于 low，那么说明该结点及它的左子树都不符合要求，我们返回对它的右结点进行修剪后的结果；
如果结点的值大于 high，那么说明该结点及它的右子树都不符合要求，我们返回对它的左子树进行修剪后的结果；
如果结点的值位于区间 [low,high], 我们将结点的左结点设为对它的左子树修剪后的结果，右结点设为对它的右子树进行修剪后的结果。

代码

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if (root == nullptr) {
            return nullptr;
        }
        if (root->val < low) {
            // 如果root（当前节点）的值小于low的值，那么应该递归处理右子树，并返回右子树符合条件的头结点。
            return trimBST(root->right, low, high);
        } else if (root->val > high) {
            // 如果root(当前节点)的值大于high的值，那么应该递归处理左子树，并返回左子树符合条件的头结点。
            return trimBST(root->left, low, high);
        }
        // 递归处理左子树，并更新根节点的左子节点
        root->left = trimBST(root->left, low, high);
        // 递归处理右子树，并更新根节点的右子节点
        root->right = trimBST(root->right, low, high);
        return root;
    }
};

复杂度分析

时间复杂度: O(n)，其中 n 为二叉树的结点数目。
空间复杂度: O(n)，递归栈最坏情况下需要 O(n) 的空间。

3、广度优先搜索（迭代法）

思路

处理头结点：
- 使用一个 while 循环，确保 root 指向的节点值在 [low, high] 范围内。
- 如果 root 的值小于 low，则移动到右子树（因为BST的右子树的值总是大于当前节点）。
- 如果 root 的值大于 high，则移动到左子树（因为BST的左子树的值总是小于当前节点）。
- 这个过程会持续，直到 root 为空或其值在 [low, high] 范围内。
检查根节点是否为空：
- 如果在处理头结点后 root 为空，说明整个树的值都不在 [low, high] 范围内，因此返回空。
修剪左子树：
- 使用一个 for 循环遍历从 root 开始的左子树。
- 对于每个节点，如果其左子节点的值小于 low，则将该左子节点替换为其右子节点（如果存在）。这样做的目的是移除所有值小于 low 的节点。
- 如果左子节点的值不小于 low，则继续向左子树遍历。
修剪右子树：
- 使用一个类似的 for 循环遍历从 root 开始的右子树。
- 对于每个节点，如果其右子节点的值大于 high，则将该右子节点替换为其左子节点（如果存在）。这样做的目的是移除所有值大于 high 的节点。
- 如果右子节点的值不大于 high，则继续向右子树遍历。
返回修剪后的根节点：
- 经过上述步骤后，得到的 root 就是修剪后的二叉搜索树的根节点，其所有节点的值都在 [low, high] 范围内。
- 返回这个 root。

代码

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        // 处理头结点，让root移动到[low, high] 范围内，注意是左闭右闭
        while (root && (root->val < low || root->val > high)) {
            if (root->val < low) {
                // 如果根节点的值小于low，则向右子树移动
                root = root->right;
            } else {
                // 如果根节点的值大于high，则向左子树移动
                root = root->left;
            }
        }
        // 如果根节点为空，则返回空
        if (root == nullptr) {
            return nullptr;
        }
        // 此时root已经在[low, high] 范围内，修剪左子树中小于 low 的节点
        for (TreeNode* node = root; node->left; ) {
            if (node->left->val < low) {
                // 如果左子节点的值小于low，则将左子节点替换为其右子节点
                node->left = node->left->right;
            } else {
                // 否则继续向左子树遍历
                node = node->left;
            }
        }
        // 此时root已经在[low, high] 范围内，修剪右子树中大于 high 的节点
        for (TreeNode* node = root; node->right; ) {  
            if (node->right->val > high) {
                // 如果右子节点的值大于high，则将右子节点替换为其左子节点
                node->right = node->right->left;
            } else {
                // 否则继续向右子树遍历
                node = node->right;
            }
        }
        // 返回调整后的根节点
        return root;
    }
};