BST二叉搜索树

2020-06-23  本文已影响0人  Kitsuna

二叉搜索树(BST)

定义

二叉搜索树又叫二叉排序树,相对于普通的二叉树,二叉搜索树规定父节点的左子节点小于父节点,右子节点大于父节点,满足这个条件的二叉树就是二叉搜索树。由于自带顺序这一特性,二叉搜索树在查找元素时的时间复杂度理想情况下可以达到O(logn),极端情况下才是O(n),因此二叉查找树主要体现了基础的查找树思路。

插入

在插入一个元素时,首先与根节点比较,元素如果大于根节点,便继续向根节点的右子树移动,如果元素小于根节点,就向根节点左子树移动,并以当前节点为父节点重新计算,直到找到空位以插入。基于这个思路,我们可以通过循环或者递归来实现。

删除

二叉搜索树的删除比较复杂,我们想一下以下三种情况:
1 被删除节点无子节点


33待删除.png

2 被删除节点只有左子树或右子树


45待删除.png
3 被删除节点同时有左右子树
48待删除.png
对于第一种情况,我们可以直接删除该节点;
删除33.png

第二种情况,删除该节点之后,由于节点的左右子树肯定满足该节点所处的条件(大于或小于该节点的父节点),这样删除策略就是直接使用该节点的左子树或右子树替代该节点;


删除45.png
第三种情况,删除掉该节点之后,节点的左右子树要怎么处理?我们应该寻找一个满足该节点条件的叶子节点来替代该节点,从而不改变树的结构。那么如何寻找这样的一个节点呢?这里引入另一个定义:

实现

public class BinarySearchTree {
    public BinarySearchTree left;
    public BinarySearchTree right;
    public Integer val;

    public BinarySearchTree(BinarySearchTree left, BinarySearchTree right, Integer val) {
        this.left = left;
        this.right = right;
        this.val = val;
    }

    public static void insert(BinarySearchTree root, int x) {
        BinarySearchTree cursor = root;
        BinarySearchTree lp = null;
        BinarySearchTree rp = null;
        while (cursor != null) {
            if (cursor.val < x) {
                lp = cursor;
                cursor = cursor.right;
                rp = null;
            } else {
                rp = cursor;
                cursor = cursor.left;
                lp = null;
            }
        }
        cursor = new BinarySearchTree(null, null, x);
        if (lp != null) {
            lp.right = cursor;
        }
        if (rp != null) {
            rp.left = cursor;
        }
    }

    public static void delete(BinarySearchTree node, int x) {
        if (node == null) {
            return;
        }
        BinarySearchTree cursor = node;
        BinarySearchTree parent = cursor;
        boolean isLeft = false;
        // 找到x所在位置
        while (cursor != null && cursor.val != x) {
            if (cursor.val < x) {
                parent = cursor;
                isLeft = false;
                cursor = cursor.right;
            } else {
                parent = cursor;
                isLeft = true;
                cursor = cursor.left;
            }
        }
        if (cursor != null) {
            // 左子树为空,直接将右子树覆盖待删除节点
            if (cursor.left == null) {
                if (cursor.right != null) {
                    cursor.val = cursor.right.val;
                    cursor.left = cursor.right.left;
                    cursor.right = cursor.right.right;
                } else {
                    if (isLeft) {
                        parent.left = null;
                    } else {
                        parent.right = null;
                    }
                }
            } else {
                // 从待删除节点左子树开始,寻找直接后继
                BinarySearchTree come = cursor.left;
                BinarySearchTree comeParent = cursor;
                while (come.right != null) {
                    comeParent = come;
                    come = come.right;
                }
                // 将直接后继的值赋给待删除节点
                cursor.val = come.val;
                // 如果直接后继没有左子树,直接删除直接后继
                if (come.left == null) {
                    comeParent.right = null;
                } else {
                    // 直接后继左子树不为空,将直接后继的左子树,覆盖直接后继
                    comeParent.right = come.left;
                }
            }
        }
    }
}
上一篇下一篇

猜你喜欢

热点阅读