双非本科准备秋招(21.1)—— 力扣二叉搜索树

刚学的二叉搜索树,做做题目巩固一下二叉搜索树的基本操作。

1、700. 二叉搜索树中的搜索

二叉搜索树的任何一个节点,都会大于左子树任意节点的值,都会小于右子树任意节点的值

java 复制代码
class Solution {
    public TreeNode searchBST(TreeNode root, int val) {
        if(root == null) return null;

        TreeNode t = root;
        while(t != null){
            if(t.val < val){
                t = t.right;
            }
            else if(val < t.val){
                t = t.left;
            }
            else{
                return t;
            }
        }
        return null;
    }
}

2、701. 二叉搜索树中的插入操作

插入操作,需要记录一下待插入节点的parent,如果parent是null,说明没有根,那么新加入的节点就是根节点,否则按照大小决定插入parent的左还是右。

java 复制代码
class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        TreeNode t = root;
        TreeNode parent = null;
        while(t != null){
            parent = t;
            if(val < t.val){
                t = t.left;
            }
            else if(t.val < val){
                t = t.right;
            }
        }
        TreeNode node = new TreeNode(val);
        if(parent == null){
            return node;
        }
        if(val < parent.val){
            parent.left = node;
        }
        else{
            parent.right = node;
        }
        return root;
    }
}

3、450. 删除二叉搜索树中的节点

分四种情况:

  • 只有左子树
  • 只有右子树
  • 叶子节点
  • 有左右子树

如果只有左子树,那么只需要把它的左孩子给它的parent节点就好,右子树同理,叶子节点一样,相当于子树为null,直接给parent也行。

如果有左右子树,那么需要用它的后继节点来替换自己,这样才能保证二叉搜索树的性质,后继节点一定不会有左子树,但可能有右子树,所以还要根据是否有右子树来判断一下。

这里用递归实现

java 复制代码
class Solution {    
    public TreeNode deleteNode(TreeNode root, int key) {
        if(root == null){
            return null;
        }
        if(key < root.val){
            root.left = deleteNode(root.left, key);
            return root;
        }
        if(root.val < key){
            root.right = deleteNode(root.right, key);
            return root;
        }
        //只有左孩子
        if(root.right == null){
            return root.left;
        }
        //只有右孩子
        if(root.left == null){
            return root.right;
        }
        //都有
        TreeNode s = root.right;
        while(s.left != null){
            s = s.left;
        }
        s.right = deleteNode(root.right, s.val);
        s.left = root.left;
        return s;
    }
}
相关推荐
JohnFF12 分钟前
48. 旋转图像
数据结构·算法·leetcode
bbc12122613 分钟前
AT_abc306_b [ABC306B] Base 2
算法
生锈的键盘21 分钟前
推荐算法实践:movielens数据集
算法
董董灿是个攻城狮22 分钟前
Transformer 通关秘籍9:词向量的数值实际上是特征
算法
代码AC不AC30 分钟前
【数据结构】队列
c语言·数据结构·学习·队列·深度讲解
林泽毅30 分钟前
SwanLab x EasyR1:多模态LLM强化学习后训练组合拳,让模型进化更高效
算法·llm·强化学习
小林熬夜学编程32 分钟前
【高并发内存池】第八弹---脱离new的定长内存池与多线程malloc测试
c语言·开发语言·数据结构·c++·算法·哈希算法
刚入门的大一新生38 分钟前
归并排序延伸-非递归版本
算法·排序算法
独好紫罗兰43 分钟前
洛谷题单3-P1980 [NOIP 2013 普及组] 计数问题-python-流程图重构
开发语言·python·算法
独好紫罗兰1 小时前
洛谷题单3-P1009 [NOIP 1998 普及组] 阶乘之和-python-流程图重构
开发语言·python·算法