Java skiplist实现优化



我正在努力实现java中的skiplist。基本上一切都很好,但put方法执行时间太长。这是我的代码,我看过教程,查看了一些人的代码,但我似乎不知道问题出在哪里(如果你测量放入100000个随机元素,需要很长时间才能工作)。

public class SkipList<K,V> {
    private final SkipListItem<K,V> head;
    private int currentLevel = 0;
    private static final Random random = new Random();
    private int height = 0;
    public SkipList() {
        this.head = new SkipListItem<>(null, null);
    }
    public V put(K key, V value) {
        return this.put(key, value, null, 0);
    }
    private V put(K key, V value, SkipListItem<K,V> previous, int level) {
        if(level > this.height) {
            for(int i=0,s=level-this.height ; i<s ; ++i) {
                this.addHeadItem();
            }
        }
        SkipListItem<K,V> addedItem = new SkipListItem<>(key, value);
        SkipListItem<K,V> insertAfter = this.findLowerItem(key, level);
        addedItem.setPrevious(insertAfter);
        if(insertAfter.getNext() != null) {
            insertAfter.getNext().setPrevious(addedItem);
            addedItem.setNext(insertAfter.getNext());
        }
        insertAfter.setNext(addedItem);
        if(previous != null) {
            addedItem.setBelow(previous);
            previous.setAbove(addedItem);
        }
        return (SkipList.random.nextBoolean()) ? this.put(key, value, addedItem, level+1) : value ;
    }
    private void addHeadItem() {
        SkipListItem<K,V> item = this.head;
        while(item.getAbove() != null) {
            item = item.getAbove();
        }
        SkipListItem<K,V> newHead = new SkipListItem<>(null,null);
        item.setAbove(newHead);
        newHead.setBelow(item);
        ++this.height;
    }
    private SkipListItem<K,V> findLowerItem(K key) {
        return this.findLowerItem(key,0);
    }
    private SkipListItem<K,V> findLowerItem(K key, int level) {
        SkipListItem<K,V> currentItem = this.head;
        for(int i=0 ; i<level ; ++i) {
            currentItem = currentItem.getAbove();
        }
        while(  currentItem.getNext() != null &&
                currentItem.getNext().getKey() != null &&
                currentItem.getNext().compareTo(key) < 0) {
            currentItem = currentItem.getNext();
        }
        return currentItem;
    }
    //...some other functions...
}

你知道为什么要花这么长时间吗?

第一眼看到的时候,只要插入新项目,就可能遍历整个列表来找到位置。这导致了线性复杂性,因此在最后插入n项需要n*n操作。

具体来说,在你的情况下,最终的数字是100亿,操作已经相当复杂了,所以我可以想象这将需要几分钟的时间来填充。

最新更新