[算法刷题打卡]Day6

最新推荐文章于 2024-07-20 09:51:47 发布

s_oftdream

最新推荐文章于 2024-07-20 09:51:47 发布

阅读量267

点赞数 10

文章标签：算法

本文链接：https://blog.csdn.net/messizq/article/details/136983635

版权

1、Leetcode-面试经典150题目12-380. O(1) 时间插入、删除和获取随机元素

思路

1.首先，变长数组可以在O（1）的时间内完成获取随机元素操作，但是无法在O(1)的时间内判断元素是否存在，因此不能再O（1）的时间内完成插入和删除操作。

2.哈希表可以在O（1）的时间内完成插入和删除，但是由于无法根据下标定位到特定元素，因此不能在 O(1)的时间内完成获取随机元素操作。

3.因此需要将变长数组和哈希表结合，变长数组存储元素，哈希表存储下标

4.对于插入操作：（1）判断val是否在哈希表中，如果不存在则插入（2）在变长数组末尾添加val （3）存入val和插入val之前数组长度index

5.对于删除操作：（1）首先判断val是否在哈希表中（2）将变长数组最后一个元素移动到下标index，在哈希表中将last下标更新为index (3)在变长数组中删除最后一个元素，在哈希表中删除 val

class RandomizedSet {
public:
    RandomizedSet() {
        srand((unsigned)time(NULL));
    }
    
    bool insert(int val) {
        if (indices.count(val)) {
            return false;
        }
        int index = nums.size();
        nums.emplace_back(val);
        indices[val] = index;
        return true;
    }
    
    bool remove(int val) {
        if (!indices.count(val)) {
            return false;
        }
        int index = indices[val];
        int last = nums.back();
        nums[index] = last;
        indices[last] = index;
        nums.pop_back();
        indices.erase(val);
        return true;
    }
    
    int getRandom() {
        int randomIndex = rand()%nums.size();
        return nums[randomIndex];
    }
private:
    vector<int> nums;
    unordered_map<int, int> indices;
};