【LeetCode】LeetCode 669.修剪二叉搜索树

NUAA_Peter

已于 2023-03-22 15:06:00 修改

阅读量1.5k

点赞数

分类专栏： # 二叉树专题文章标签： leetcode 算法职场和发展二叉搜索树

于 2022-07-26 00:28:39 首次发布

本文链接：https://blog.csdn.net/peterwanye/article/details/125986092

版权

二叉树专题专栏收录该内容

25 篇文章 0 订阅

订阅专栏

LeetCode 669.修剪二叉搜索树

题目链接：https://leetcode.cn/problems/trim-a-binary-search-tree/

题目

解答

递归法

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

        if(root->val > high){
            TreeNode* left = trimBST(root->left, low, high);
            return left;
        }

        root->left = trimBST(root->left, low, high);
        root->right = trimBST(root->right, low, high);
        return root;
    }
};

迭代法

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

        while(root != NULL && (root->val < low || root->val > high)){
            if(root->val < low)  root = root->right;
            else  root = root->left;
        }

        TreeNode* cur = root;
        while(cur != NULL){
            while(cur->left != NULL && cur->left->val < low){
                cur->left = cur->left->right;
            }
            cur = cur->left;
        }

        cur = root;

        while(cur != NULL){
            while(cur->right != NULL && cur->right->val > high){
                cur->right = cur->right->left;
            }
            cur = cur->right;
        }
        return root;
    }
};