二叉树 | 31 669. 修剪二叉搜索树

本文记录的是刷题过程中的重要概念和笔记。如有侵权，请联系删除。

669. 修剪二叉搜索树

力扣题目链接(opens new window)

给定一个二叉搜索树，同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树，使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点，所以结果应当返回修剪好的二叉搜索树的新的根节点。

669.修剪二叉搜索树

递归

三要素分析

单层函数：

• 自己要改
• 儿子要改

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if (root == nullptr ) return nullptr;
        if (root->val < low) {
            TreeNode* right = trimBST(root->right, low, high); // 寻找符合区间[low, high]的节点
            return right;
        }
        if (root->val > high) {
            TreeNode* left = trimBST(root->left, low, high); // 寻找符合区间[low, high]的节点
            return left;
        }
        root->left = trimBST(root->left, low, high); // root->left接入符合条件的左孩子
        root->right = trimBST(root->right, low, high); // root->right接入符合条件的右孩子
        return root;
    }
};

迭代

比较抽象

• 根节点要转到符合条件的接顶处，这个过程中舍弃到点节点其实都是不符合条件的，不用管它。
• 然后处理此时更节点的左边节点。如果出现不符合条件的就修剪掉再处理右边。

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

        TreeNode *cur = root;
        // 处理头结点，让root移动到[L, R] 范围内，注意是左闭右闭
        while (cur && (cur->val > high || cur->val < low))
        {
            if (cur->val > high)
                cur = cur->left;
            else
                cur = cur->right;
        }
        root = cur;

        if (cur)
        { // cur非空才可访问，否则会非法访问内存
            // 此时root已经在[L, R] 范围内，处理左孩子元素小于L的情况
            while (cur->left)
            {
                if (cur->left->val < low)
                {
                    cur->left = cur->left->right;
                    continue;
                }
                cur = cur->left;
            }
            // 此时root已经在[L, R] 范围内，处理右孩子大于R的情况
            cur = root;
            while (cur->right)
            {
                if (cur->right->val > high)
                {
                    cur->right = cur->right->left;
                    continue;
                }
                cur = cur->right;
            }
        }
        return root;
    }
};

非空才可以访问，注意加判断语句，否则会非法访问内存