write a skiplist by java

import java.util.ArrayList;

public class SkipList {
    // Node of the SkipList
    public static class SkipListNode<k extends comparable>, V> {
        public K key;
        public V value;
        public ArrayList<skiplistnode v>> nextNodes;

        public SkipListNode(K key, V value) {
            this.key = key;
            this.value = value;
            nextNodes = new ArrayList<skiplistnode v>>();
        }
    }

    // SkipList
    public static class SkipListMap<k extends comparable>, V> {
        public static final double PROBABILITY = 0.5; // Probability for level generation
        public SkipListNode<k v> head;
        public int maxLevel; // Maximum level in the SkipList
        public int size;

        public SkipListMap() {
            // The head is the leftmost "platform"
            this.head = new SkipListNode(null, null);
            head.nextNodes.add(null); // Level 0
            this.size = 0;
            this.maxLevel = 0;
        }

        // Add a node to the SkipList
        public void put(K key, V value) {
            if (key == null) {
                return;
            }
            // Check if the key already exists in the SkipList, update the value if so
            // Method: Find the rightmost node less than the key at the bottom level
            SkipListNode<k v> less = mostRightLessNodeInTree(key);
            // less.nextNodes.get(0) --
            SkipListNode<k v> find = less.nextNodes.get(0);
            if (find.key.compareTo(key) == 0) {
                find.value = value;
            } else {
                // Generate a random level for the new node
                int newNodeLevel = 0;
                while (Math.random()  maxLevel) {
                    maxLevel++;
                    head.nextNodes.add(null);
                }
                SkipListNode<k v> newNode = new SkipListNode(key, value);
                for (int i = 0; i  pre = head;
                while (level >= 0) {
                    // Find the predecessor node at the current level
                    pre = mostRightLessNodeInLevel(pre, key, level);
                    // Insert the new node between the predecessor and its successor
                    if (level  pre = head;
            while (level >= 0) {
                // Find the predecessor node at the current level
                pre = mostRightLessNodeInLevel(pre, key, level);
                // Remove the node
                SkipListNode<k v> next = pre.nextNodes.get(level);
                if (next != null && next.key.compareTo(key) == 0) {
                    pre.nextNodes.set(level, next.nextNodes.get(level));
                }
                // If a level has only the head node left, remove this level
                if (level != 0 && pre == head && pre.nextNodes.get(level) == null) {
                    head.nextNodes.remove(level);
                    maxLevel--;
                }
                level--;
            }
        }

        // Start from the top level and traverse down to find the rightmost node less than the key at level 0
        public SkipListNode<k v> mostRightLessNodeInTree(K key) {
            if (key == null) {
                return null;
            }
            int level = maxLevel;
            SkipListNode<k v> cur = head;
            while (level >= 0) {
                cur = mostRightLessNodeInLevel(cur, key, level);
                level--;
            }
            return cur;
        }

        // At a specific level, find the rightmost node less than the key
        public SkipListNode<k v> mostRightLessNodeInLevel(SkipListNode<k v> cur, K key, int level) {
            if (key == null) {
                return null;
            }
            SkipListNode<k v> pre = null;
            cur = cur.nextNodes.get(level);
            while (cur.key.compareTo(key)  less = mostRightLessNodeInTree(key);
            SkipListNode<k v> find = less.nextNodes.get(0);
            return find != null && find.key.compareTo(key) == 0;
        }
    }
}

</k></k></k></k></k></k></k></k></k></k></k></k></skiplistnode></skiplistnode></k>

The above is the detailed content of write a skiplist by java. For more information, please follow other related articles on the PHP Chinese website!