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

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

相关推荐
qq_513970444 分钟前
力扣 hot100 Day37
算法·leetcode
掉鱼的猫5 分钟前
Java MCP 实战:构建跨进程与远程的工具服务
java·openai·mcp
编程零零七23 分钟前
Python巩固训练——第一天练习题
开发语言·python·python基础·python学习·python练习题
不見星空24 分钟前
leetcode 每日一题 1865. 找出和为指定值的下标对
算法·leetcode
我爱Jack34 分钟前
时间与空间复杂度详解:算法效率的度量衡
java·开发语言·算法
米饭「」36 分钟前
C++AVL树
java·开发语言·c++
Zonda要好好学习44 分钟前
Python入门Day4
java·网络·python
SimonKing1 小时前
告别传统读写!RandomAccessFile让你的Java程序快人一步
java·后端·程序员
Little-Hu1 小时前
QML TextEdit组件
java·服务器·数据库
心愿许得无限大1 小时前
Qt 常用界面组件
开发语言·c++·qt