O(1) 时间插入、删除和获取随机元素

最新推荐文章于 2024-08-06 10:27:31 发布

tuzittttt

最新推荐文章于 2024-08-06 10:27:31 发布

阅读量258

点赞数

分类专栏：力扣每日刷题文章标签： c++

本文链接：https://blog.csdn.net/songyaxuanwoai/article/details/124166134

版权

力扣每日刷题专栏收录该内容

20 篇文章 0 订阅

订阅专栏

该博客主要介绍了一种数据结构设计，通过结合unordered_map（哈希表）和vector（变长数组），实现在常数时间内进行插入、删除和随机获取元素的操作。具体实现中，哈希表用于存储元素值与其在数组中的位置映射，而数组则存储实际的元素值。在插入和删除操作中，当元素存在时，会进行位置的调整以保持数组的连续性。此外，getRandom()方法通过随机数生成器确保所有元素被等概率随机获取。

摘要由CSDN通过智能技术生成

380. O(1) 时间插入、删除和获取随机元素

题目链接：https://leetcode-cn.com/problems/insert-delete-getrandom-o1/
思路：要完成 $O （ 1 ）$ 的查找第一时间就可以想到哈希表，但是由于哈希表只能添加 $O (1)$ 添加，删除和查找，由于无法确定哈希的长度，还需要用一个变长数组来辅助储存，哈希表用来储存每个val的位置，vector用来存储val的值

代码：

class RandomizedSet {
    unordered_map<int, int> hash;//第一个存val,第二个存位置
    vector<int> a;//下标为位置，为值
public:
    RandomizedSet() {
    }
    
    bool insert(int val) {
        if(hash.find(val) == hash.end()){
            a.push_back(val);
            int mid = a.size() - 1;
            hash[val] = mid;
            return true;
        }
        else return false;

    }
    
    bool remove(int val) {
        if(hash.find(val) != hash.end()){
            int mid = hash[val];//取出val的下标
            int index = a.size() - 1;//
            hash[a[index]] = mid;//将index对应的下标变成val的下标
            a[mid] = a[index];//将mid上存储的树变成index上存储的数据
            a.pop_back();//取出最后一个数
            hash.erase(val);//删除最后一个s
            return true;
        }
        else return false;


    }
    
    int getRandom() {
        if(a.size() == 0) return -1;
        int index = rand() % a.size();//随机生成一个数与a的长度取模，保证每个元素同等的机会被选到
        return a[index];

        
        

    }
};

/**
 * Your RandomizedSet object will be instantiated and called as such:
 * RandomizedSet* obj = new RandomizedSet();
 * bool param_1 = obj->insert(val);
 * bool param_2 = obj->remove(val);
 * int param_3 = obj->getRandom();
 */