[LeetCode] 380.常数时间插入、删除和获取随机元素（Medium）C语言题解

题目

• 设计一个支持在平均 时间复杂度 O(1) 下，执行以下操作的数据结构。
• insert(val)：当元素 val 不存在时，向集合中插入该项。
• remove(val)：元素 val 存在时，从集合中移除该项。
• getRandom：随机返回现有集合中的一项。每个元素应该有相同的概率被返回。

---

示例

①示例1

---

说明

①数据范围(自测)

• -2147483648 ~ 2147483647

②相关话题

• 设计
• 哈希表
• 数组

③相似题目

• 381. O(1) 时间插入、删除和获取随机元素 - 允许重复 — 力扣网
• 381. Insert Delete GetRandom O(1) - Duplicates allowed — leetcode

④题目地址

• 380. 常数时间插入、删除和获取随机元素 — 力扣网
• 380. Insert Delete GetRandom O(1) — leetcode

---

解题方法

①哈希表

• 因为数据范围太大，无法用数组模拟的哈希表求解，但是可以使用 uthash。
• uthash 是一个用 C 语言编写的开源库，使用宏实现了哈希表的增删改查等功能。
• 虽然使用 uthash 可以实现常数时间插入、删除元素，但是无法实现常数时间获取随机元素。
• 所以我们还要再使用一个数组保存所有已插入的元素，以便于获取随机元素。并且要保证数组范围内的值都是有效的，即在插入元素时记录其数组下标值，以便在删除时，让数组中的最后一个元素填补这个删除空档。
• 时间复杂度：O(1)。
• 空间复杂度：O(N)。

---

代码详解

• 哈希表(uthash)

struct hash {
    int key;
    // 保存key在数组arr中的下标值——便于删除元素。
    int value;
    UT_hash_handle hh; 
};

typedef struct {
    // 数组(存放所有已插入的元素)——便于O(1)时间获取随机元素。
    int* arr;
    // 记录数组中元素的个数。
    int size;
    // 哈希表——便于O(1)时间插入和删除元素。
    struct hash* hashTable; 
} RandomizedSet;

/** Initialize your data structure here. */
RandomizedSet* randomizedSetCreate() {
    RandomizedSet* r = malloc(sizeof(RandomizedSet));
    r->arr = malloc(sizeof(int)*5000);
    r->size = 0;
    r->hashTable = NULL;

    return r;
}

bool randomizedSetInsert(RandomizedSet* obj, int val) {
    struct hash *h; 
    HASH_FIND_INT(obj->hashTable, &val, h);
    
    if (!h) {
        h = malloc(sizeof(struct hash));
        h->key = val;
        // 存入数组、并记录其在数组中存储的下标值——便于删除元素。
        obj->arr[obj->size] = val;
        h->value = obj->size;
        obj->size++;
        HASH_ADD_INT(obj->hashTable, key, h);
        
        return true;
    }
    
    return false;
}

bool randomizedSetRemove(RandomizedSet* obj, int val) {
    struct hash *h1, *h2; 
    HASH_FIND_INT(obj->hashTable, &val, h1);
    
    // 如果存在，则将数组中最后一个元素移到此元素的位置，保证0~obj->size的数组值有效。
    if (h1) {
        obj->arr[h1->value] = obj->arr[obj->size-1]; 
        HASH_FIND_INT(obj->hashTable, &obj->arr[obj->size-1], h2);
        h2->value = h1->value;
        obj->size--;
        HASH_DEL(obj->hashTable, h1);
        
        return true;
    }
    
    return false;
}

// 获取随机元素。
int randomizedSetGetRandom(RandomizedSet* obj) {
    return obj->arr[rand()%obj->size;];
}

void randomizedSetFree(RandomizedSet* obj) {
    free(obj->arr);
    free(obj);
}

---

附录

• 我的个人博客：messi1002.top
• 如有错误或疑惑之处 请联系 wjymessi@163.com
• 所有题目解答：fork me on github