Java 实现二叉搜索树 代码

新建文件
创建TreeNode类,实例化

直接在BinarySearchTree类里面写就可以

java 复制代码
static class TreeNode {
        public int key;
        public TreeNode left;
        public TreeNode right;

        TreeNode(int key) {
            this.key = key;
        }
    }

    public TreeNode root;
插入节点 insert
java 复制代码
 public boolean insert(int key) {
        TreeNode node = new TreeNode(key);
        if(root==null){
            //检查是否为空树
            root=node;
            return true;//表示成功插入
        }
        TreeNode cur=root;//读取根节点
        TreeNode parent=null;
        //查找合适的位置 如果已经有了则返回false
        while(cur!=null){
            if(cur.key<key){
                parent=cur;
                cur=cur.right;
            } else if (cur.key>key) {
                parent=cur;
                cur=cur.left;
            }else{
                return false;
            }
        }
        //插入
        if(key< parent.key){
            parent.left=node;
        }else{
            parent.right=node;
        }
        return true;
    }

找到则返回该节点,没找到返回null

java 复制代码
public TreeNode search(int key) {
        if(root==null) return null;
        TreeNode cur=root;
        while(cur!=null){
            if(cur.key==key){
                return cur;
            }else if(cur.key>key){
                cur=cur.left;
            }else{
                cur=cur.right;
            }
        }
        return null;
    }
删除节点 remove

找到节点并删除,成功则返回true,失败false

java 复制代码
 public boolean remove(int key) {
        TreeNode cur=root;
        TreeNode parent=null;
        while(cur!=null){
            if(cur.key<key){
                parent=cur;
                cur=cur.right;
            } else if (cur.key>key) {
                parent=cur;
                cur=cur.left;
            }else{
                removeNode(parent,cur);//进行删除
                return true;
            }
        }
        return false;
    }

注意,这里删除又写了一个自定义函数。已至删除的位置和其双亲结点,如何让其删除呢?

如果左右子节点都在,则用左子树的最大值或者右子树的最小值替换即可,下面给出用右子树的最小值替换的方法。

java 复制代码
public  void removeNode(TreeNode parent,TreeNode cur){
        //在子树部分删除cur节点后重新构建子搜索二叉数
        if(cur.left==null){
            if(cur==root) root=cur.right;
            if(cur==parent.left) parent.left=cur.right;
            if(cur==parent.right) parent.right=cur.right;
        }else if(cur.right==null){
            if(cur==root) root=cur.left;
            if(cur==parent.left) parent.left=cur.left;
            if(cur==parent.right) parent.right=cur.left;
        }else{
            TreeNode target=cur.right;
            TreeNode targetParent=cur;
            while(target.left!=null){
                targetParent=target;
                target=target.left;
            }
            cur.key= target.key;
            if(target==targetParent.left) {
                targetParent.left = target.right;
            }else{
                targetParent.right = target.right;
            }
        }
    }

如有什么没讲清楚的地方,欢迎留言探讨!

相关推荐
学习前端的小z4 分钟前
【前端】深入理解 JavaScript 逻辑运算符的优先级与短路求值机制
开发语言·前端·javascript
神仙别闹11 分钟前
基于C#和Sql Server 2008实现的(WinForm)订单生成系统
开发语言·c#
XINGTECODE12 分钟前
海盗王集成网关和商城服务端功能golang版
开发语言·后端·golang
alphaTao15 分钟前
LeetCode 每日一题 2024/11/18-2024/11/24
算法·leetcode
天天扭码18 分钟前
五天SpringCloud计划——DAY2之单体架构和微服务架构的选择和转换原则
java·spring cloud·微服务·架构
程序猿进阶18 分钟前
堆外内存泄露排查经历
java·jvm·后端·面试·性能优化·oom·内存泄露
FIN技术铺23 分钟前
Spring Boot框架Starter组件整理
java·spring boot·后端
kitesxian24 分钟前
Leetcode448. 找到所有数组中消失的数字(HOT100)+Leetcode139. 单词拆分(HOT100)
数据结构·算法·leetcode
zwjapple29 分钟前
typescript里面正则的使用
开发语言·javascript·正则表达式
小五Five30 分钟前
TypeScript项目中Axios的封装
开发语言·前端·javascript