以下是一个简单的Java实现红黑树算法的代码:

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

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

    private Node root;

    private class Node {
        T key;
        Node left, right;
        boolean color;

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

    public void insert(T key) {
        root = insert(root, key);
        root.color = BLACK;
    }

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

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

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

    private boolean isRed(Node node) {
        if (node == null) {
            return false;
        }
        return node.color == RED;
    }

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

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

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

该实现使用了红黑树的基本操作,包括插入和旋转。在插入节点时,会进行颜色翻转和旋转操作以保持树的平衡和满足红黑树的性质。具体实现细节可以参考代码注释

用java写一个红黑树算法

原文地址: https://www.cveoy.top/t/topic/frBh 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录