669. Trim a Binary Search Tree

xingkong1992

于 2018-01-11 12:00:59 发布

阅读量129

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode学习

本文链接：https://blog.youkuaiyun.com/u012778718/article/details/79032630

leetcode学习专栏收录该内容

146 篇文章

订阅专栏

本文介绍了一种修剪二叉搜索树的方法，通过递归算法确保树中所有节点的值位于指定范围内，对于超出范围的节点进行删除操作。该算法保持了二叉搜索树的基本特性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这里写图片描述

    TreeNode* trimBST(TreeNode* root, int L, int R) {
        preOrder(root,L,R);
        return root;
    }
    void preOrder(TreeNode* &node,int l,int r){
        if(!node) return ;
        if(node->val<l)
            node=node->right;
        if(node->val>r)
            node=node->left;
        if(node->val==l)
            node->left=NULL;
        if(node->val==r)
            node->right=NULL;       
        if(node->val>=l&&node->val<=r){
            preOrder(node->left,l,r);
            preOrder(node->right,l,r);
        }else{
            preOrder(node,l,r);
            preOrder(node,l,r);
        }      
    }

    TreeNode* trimBST(TreeNode* root, int L, int R) {
        if(!root) return NULL;
        if(root->val>R) return trimBST(root->left,L,R);
        if(root->val<L) return trimBST(root->right,L,R);
        root->left=trimBST(root->left,L,R);
        root->right=trimBST(root->right,L,R);
        return root;
    }

    TreeNode* trimBST(TreeNode* root, int L, int R,bool top=true) {
        if(!root) return NULL;
        root->left=trimBST(root->left,L,R,false);
        root->right=trimBST(root->right,L,R,false);
        if(root->val>=L&&root->val<=R)
            return root;
        auto res=(root->val<L)?root->right:root->left;
        if(!top)
          delete root;
        return res;
    }