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;
            }
        }
    }

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

相关推荐
迎風吹頭髮13 小时前
Linux服务器编程实践58-getnameinfo函数:通过socket地址获取主机名与服务名
开发语言·数据库·php
一吃就胖的13 小时前
【给服务器安装服务器安装nacos】
java·运维·服务器
爱和冰阔落13 小时前
【C++多态】虚函数/虚表机制与协变 、override和final关键字全解析
开发语言·c++·面试·腾讯云ai代码助手
码住懒羊羊13 小时前
【C++】stack|queue|deque
java·开发语言·c++
hdsoft_huge13 小时前
第六章 Kettle(PDI)解锁脚本组件:数据处理的可编程利器
java·大数据·etl
一只专注api接口开发的技术猿13 小时前
容器化与调度:使用 Docker 与 K8s 管理分布式淘宝商品数据采集任务
开发语言·前端·数据库
夏鹏今天学习了吗13 小时前
【LeetCode热题100(47/100)】路径总和 III
算法·leetcode·职场和发展
smj2302_7968265213 小时前
解决leetcode第3721题最长平衡子数组II
python·算法·leetcode
聪明的笨猪猪13 小时前
Java JVM “内存(1)”面试清单(含超通俗生活案例与深度理解)
java·经验分享·笔记·面试
m0_6265352014 小时前
力扣题目练习 换水问题
python·算法·leetcode