跳表简单实现

跳表

class Skiplist {
    static const int SKIPLIST_P_VAL = RAND_MAX / 2, MAX_LEVEL = 16; // RAND_MAX是rand()所能达到的最大值。
public:
    struct Node{
        int val;
        vector<Node *> next;
        Node(int val, int size = MAX_LEVEL): val(val), next(size){}
    };
    Node head;
    int maxlevel = 1; // 记录当前最高的level

    Skiplist():head(INT_MIN, MAX_LEVEL) {
    }
    bool search(int target) {
        // 在_search后就看第一层的pre的后面一个节点val是否等于target
        auto prevs = _search(target);
        return prevs[0]->next[0] && prevs[0]->next[0]->val == target;
    }

    vector<Node *> _search(int key){
        // 此函数是跳表的核心，实现的功能是，寻找key所经历的路径。返回路径上的前置节点
        Node * cur = &head;
        vector<Node *> prevs(MAX_LEVEL);
        for(int i = maxlevel - 1; i >= 0; i--){
            // 从最顶层开始
            while(cur->next[i] && cur->next[i]->val < key){
                // 一直寻找到刚好大于key的节点的前置节点
                cur = cur->next[i];
            }
            prevs[i] = cur;
        }
        return prevs;
    }
    
    void add(int num) {
        auto prevs = _search(num); // 在当前的跳表中， num在各级跳表中的前置节点
        int level = random_level(); // level的索引是从1开始的

        // 更新maxlevel, prevs
        if(level > maxlevel){
            // 更新超出maxlevel的prevs，前置节点都为head
            for(int i = maxlevel; i < level; i++){
                prevs[i] = &head;  
            }
            maxlevel = level;
        }

        // 创建当前节点
        Node * cur = new Node(num, level);

        // 插入在prevs节点后, 从pre->nxd到pre->cur->nx
        for(int i = level - 1; i >= 0; i--){
            cur->next[i] = prevs[i]->next[i];
            prevs[i]->next[i] = cur;
        }

    }
    
    bool erase(int num) {
        auto prevs = _search(num);

        // 如果num不存在跳表中
        if(!prevs[0]->next[0] || prevs[0]->next[0]->val != num){
            return false;
        }

        Node * del = prevs[0]->next[0];
        // 删除prevs后的节点
        for(int i = 0; i < maxlevel; i++){
            if(prevs[i]->next[i] == del){
                prevs[i]->next[i] = del->next[i];
            }
        }
        delete del;

        // 更新maxlevel
        while(maxlevel > 1 && !head.next[maxlevel - 1]){
            maxlevel--;
        }
        return true;

    }
    

    static int random_level(){
        int level = 1;
        while(rand() < SKIPLIST_P_VAL && level < MAX_LEVEL){
            level++;
        }
        return level; // 此处返回的level的索引是从1开始的
    }
};