二叉查找树代码实现

一、定义

  1. 查找二叉树又叫二叉排序树,如果同时满足以下性质:
    1. 左右子树都是二叉树
    2. 如果左子树不为空,那么左子树上面的所有节点的关键字都比根节点的关键字小;
    3. 如果右子树不为空,那么右子树上面的所有节点的关键字都比跟节点的关键字大;
  2. 好处:方便查找,可以提高查找效率,查找二叉树越平衡,查找的效率的越稳定,效率越高,这种树就是平衡二叉树

二、代码实现

public class SearchBinaryTree {

    private TreeNode root;//跟节点

    /**
     * 创建查找二叉树,如果关键字比根节点的关键字大,就往右边插入,
     * 如果关键字比根节点小,就往左边插入
     * @param data 带插入的数据
     * @return
     */
    public TreeNode put(int data) {
        TreeNode node = null;
        TreeNode parent = null;
        //首先判断根节点,如果根节点为空,则新创建的节点为根节点
        if (root == null) {
            node = new TreeNode(0, data);
            root = node;
            return node;
        } else {
            //如果不是根节点,就需要不停地遍历已有的节点,找到满足条件的节点的位置,然后插入节点
            //和根节点进行比较,首先要拿到根节点
            node = root;
            while (node != null) {
                //把当前结点制定为父节点,然后比较左右孩子和当前节点的大小
                parent = node;
                //如果要插入的数据比当前结点要大,就需要往右子树上找
                if (data > node.data) {
                    node = node.rightChild;
                } else if (data < node.data) {
                    node = node.leftChild;
                } else {
                    return node;
                }
            }
            //while循环结束,就表示找到了要插入的地方,
            // 接下来就是创建节点,然后把这个节点挂在指定的位置
            //角标默认是0
            node = new TreeNode(0, data);
            //节点创建完毕,需要判断放在父节点的左边还是右边
            if (data < parent.data) {
                parent.leftChild = node;
            } else {
                parent.rightChild = node;
            }
            //指定这个节点的父节点
            node.parent = parent;
            return node;
        }
    }

    /**
     * 删除二叉查找树中的某个节点,
     * <strong>因为是二叉查找树,所以是删除某个节点后,需要对剩下的节点进行
     * 重新排序,使之依然是二叉查找树
     * </strong>
     * @param data
     */
    public void deleteNode(int data) {
        //要想删除某个节点,必须先找到这个节点
        TreeNode node = searchNode(data);
        //如果node为空,表示没有找到
        if (node == null) {
            throw new NoSuchElementException("没有指定的节点");
        } else {
            delete(node);
        }
    }

    /**
     * 删除节点
     * @param node 要删除的节点
     */
    private void delete(TreeNode node) {
        if (node == null) {
            throw new NoSuchElementException("没有这个节点");
        } else {
            //如果当前节点不为空,就需要判断这个节点的父节点,左右孩子节点是否为空
            TreeNode parent = node.parent;
            //如果 当前节点为叶子节点,直接删除
            if (node.leftChild == null && node.rightChild == null) {
                //需要判断是父亲的左儿子还是右儿子
                if (parent.leftChild == node) {
                    parent.leftChild = null;
                    node.parent = null;
                } else {
                    parent.rightChild = null;
                    node.parent = null;
                }
            } else if (node.leftChild != null && node.rightChild == null) {
                //如果被删除的节点有左孩子没有右孩子
                //这是还需要判断这个节点是在父节点的左子树上,还是在右子树上
                //如果这个节点在左子树上,就需要把这个节点的左孩子放在parent的左子树上
                if (parent.leftChild == node) {
                    parent.leftChild = node.leftChild;
                } else {
                    parent.rightChild = node.leftChild;
                }
            } else if (node.leftChild == null && node.rightChild != null) {
                //如果被删除的节点有右孩子而没有左孩子
                if (parent.leftChild == node) {
                    parent.leftChild = node.rightChild;
                } else {
                    parent.rightChild = node.rightChild;
                }
            } else if (node.leftChild != null && node.rightChild != null) {
                //如果被删除的节点既有左孩子又有右孩子
                //找后继节点
                TreeNode next = getNextNode(node);
                delete(next);
                node.data = next.data;
            }
        }
    }

    /**
     * 找到某个节点的后继节点
     * @param node
     * @return
     */
    private TreeNode getNextNode(TreeNode node) {
        if (node == null) {
            return null;
        } else {
            //如果这个节点的右子树不为空,就在右子树里面找到key最小的节点
            if (node.rightChild != null) {
                return getMinTreeNode(node);
            } else {
                TreeNode parent = node.parent;
                while (parent == null && node == parent.rightChild) {
                    node = parent;
                    parent = parent.parent;
                }
                return parent;
            }
        }
    }

    /**
     * 找到这个节点的子树下面key最小的节点,最小的节点一定在该节点的左子树上的最末端
     * @param node
     * @return
     */
    private TreeNode getMinTreeNode(TreeNode node) {
        if (node == null) {
            return null;
        } else {
            while (node.leftChild != null) {
                node = node.leftChild;
            }
        }
        return node;
    }

    private TreeNode searchNode(int data) {
        TreeNode node;
        //先从根节点开始找
        node = root;
        if (node == null) {
            return null;
        } else {
            while (node != null && data != node.data) {
                if (data > node.data) {
                    node = node.rightChild;
                } else if (data < node.data) {
                    node = node.leftChild;
                } else {
                    return node;
                }
            }
            //当while循环结束后,就表示已经遍历完了整个二叉查找树,
            // 有可能找到,有可能找不到,即node有可能为null
            return node;
        }
    }

    /**
     * 中序遍历二叉查找树,如果树构建的正确,中序遍历出来应该是升序的
     * @param node
     */
    public void midOrderTraverse(TreeNode node) {
        midOrderTraverse(node.leftChild);
        System.out.print(node.data + " ");
        midOrderTraverse(node.rightChild);
    }

    private static final class TreeNode {
        private int key;//二叉查找树的关键字,通过这个关键字对节点进行排序
        private TreeNode leftChild;
        private TreeNode rightChild;
        private TreeNode parent;
        private int data;

        public TreeNode(int key, int data) {
            this.key = key;
            this.data = data;
            this.leftChild = null;
            this.rightChild = null;
            this.parent = null;
        }

        public int getKey() {
            return key;
        }

        public void setKey(int key) {
            this.key = key;
        }

        public TreeNode getLeftChild() {
            return leftChild;
        }

        public void setLeftChild(TreeNode leftChild) {
            this.leftChild = leftChild;
        }

        public TreeNode getRightChild() {
            return rightChild;
        }

        public void setRightChild(TreeNode rightChild) {
            this.rightChild = rightChild;
        }

        public TreeNode getParent() {
            return parent;
        }

        public void setParent(TreeNode parent) {
            this.parent = parent;
        }

        public int getData() {
            return data;
        }

        public void setData(int data) {
            this.data = data;
        }
    }
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,271评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,275评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,151评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,550评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,553评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,559评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,924评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,580评论 0 257
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,826评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,578评论 2 320
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,661评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,363评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,940评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,926评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,156评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,872评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,391评论 2 342

推荐阅读更多精彩内容