红黑树实现-查找、插入、删除

红黑树是一种自平衡二叉查找树,它保证了插入、删除和查找操作的最坏情况时间复杂度为 O(log n)。红黑树的节点有两种颜色,红色和黑色,同时满足以下性质:

  1. 根节点为黑色;
  2. 所有叶子节点都是黑色;
  3. 如果一个节点是红色的,则它的两个子节点都是黑色的;
  4. 从任意一个节点到其每个叶子节点的所有路径都包含相同数目的黑色节点;
  5. 每个节点的左子树和右子树都是红黑树。

红黑树的基本操作:

  1. 插入:首先按照二叉搜索树的方式插入节点,并将该节点的颜色设置为红色。如果插入的节点违反了红黑树的性质,就要进行旋转和变色操作来修复树的结构,使它重新满足红黑树的五个性质;
  2. 删除:首先按照二叉搜索树的方式删除节点,并将其子树重新连接起来。如果删除的节点是黑色节点,就可能破坏红黑树的性质,需要进行旋转和变色操作来修复树的结构;
  3. 查找:与二叉搜索树相同,按照节点的值进行比较,进行查找操作。

红黑树的应用:

  1. C++ STL中的map和set都是通过红黑树实现的;
  2. 数据库中的索引结构,如B+树,也是基于红黑树的改进实现的;
  3. Linux内核中的进程调度、文件系统等也利用了红黑树的特性;
  4. jdk8中hashmap链表长度超过8,转为红黑树。

方法包括:查找,删除,插入

其中,插入节点和删除节点都需要进行旋转和颜色翻转等操作来保证红黑树的性质。查找节点则是一个简单的二叉查找

下面是 Java 实现红黑树的代码:

public class RedBlackTree<T extends Comparable<T>> {

    private Node<T> root;

    private static final boolean RED = true;
    private static final boolean BLACK = false;

    private static class Node<T> {
        T value;
        Node<T> left, right;
        boolean color;

        Node(T value, boolean color) {
            this.value = value;
            this.color = color;
        }
    }

    // 插入节点
    public void insert(T value) {
        root = insert(root, value);
        root.color = BLACK;
    }

    private Node<T> insert(Node<T> node, T value) {
        if (node == null) {
            return new Node<>(value, RED);
        }

        int cmp = value.compareTo(node.value);
        if (cmp < 0) {
            node.left = insert(node.left, value);
        } else if (cmp > 0) {
            node.right = insert(node.right, value);
        } else {
            node.value = value;
        }

        if (isRed(node.right) && !isRed(node.left)) {
            node = rotateLeft(node);
        }
        if (isRed(node.left) && isRed(node.left.left)) {
            node = rotateRight(node);
        }
        if (isRed(node.left) && isRed(node.right)) {
            flipColors(node);
        }

        return node;
    }

    // 删除节点
    public void delete(T value) {
        if (root == null) {
            return;
        }

        if (!isRed(root.left) && !isRed(root.right)) {
            root.color = RED;
        }

        root = delete(root, value);

        if (root != null && !isRed(root.left) && !isRed(root.right)) {
            root.color = BLACK;
        }
    }

    private Node<T> delete(Node<T> node, T value) {
        if (value.compareTo(node.value) < 0) {
            if (!isRed(node.left) && !isRed(node.left.left)) {
                node = moveRedLeft(node);
            }
            node.left = delete(node.left, value);
        } else {
            if (isRed(node.left)) {
                node = rotateRight(node);
            }
            if (value.compareTo(node.value) == 0 && node.right == null) {
                return null;
            }
            if (!isRed(node.right) && !isRed(node.right.left)) {
                node = moveRedRight(node);
            }
            if (value.compareTo(node.value) == 0) {
                Node<T> min = findMin(node.right);
                node.value = min.value;
                node.right = deleteMin(node.right);
            } else {
                node.right = delete(node.right, value);
            }
        }

        return balance(node);
    }

    // 查找节点
    public Node<T> find(T value) {
        Node<T> node = root;
        while (node != null) {
            int cmp = value.compareTo(node.value);
            if (cmp < 0) {
                node = node.left;
            } else if (cmp > 0) {
                node = node.right;
            } else {
                return node;
            }
        }
        return null;
    }

    private boolean isRed(Node<T> node) {
        return node != null && node.color == RED;
    }

    private Node<T> rotateLeft(Node<T> node) {
        Node<T> x = node.right;
        node.right = x.left;
        x.left = node;
        x.color = node.color;
        node.color = RED;
        return x;
    }

    private Node<T> rotateRight(Node<T> node) {
        Node<T> x = node.left;
        node.left = x.right;
        x.right = node;
        x.color = node.color;
        node.color = RED;
        return x;
    }

    private void flipColors(Node<T> node) {
        node.color = RED;
        node.left.color = BLACK;
        node.right.color = BLACK;
    }

    private Node<T> moveRedLeft(Node<T> node) {
        flipColors(node);
        if (isRed(node.right.left)) {
            node.right = rotateRight(node.right);
            node = rotateLeft(node);
            flipColors(node);
        }
        return node;
    }

    private Node<T> moveRedRight(Node<T> node) {
        flipColors(node);
        if (isRed(node.left.left)) {
            node = rotateRight(node);
            flipColors(node);
        }
        return node;
    }

    private Node<T> findMin(Node<T> node) {
        while (node.left != null) {
            node = node.left;
        }
        return node;
    }

    private Node<T> deleteMin(Node<T> node) {
        if (node.left == null) {
            return null;
        }

        if (!isRed(node.left) && !isRed(node.left.left)) {
            node = moveRedLeft(node);
        }

        node.left = deleteMin(node.left);
        return balance(node);
    }

    private Node<T> balance(Node<T> node) {
        if (isRed(node.right)) {
            node = rotateLeft(node);
        }
        if (isRed(node.left) && isRed(node.left.left)) {
            node = rotateRight(node);
        }
        if (isRed(node.left) && isRed(node.right)) {
            flipColors(node);
        }
        return node;
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值