1. 題目

設(shè)計一個支持在平均 時間復(fù)雜度 O(1) 下，執(zhí)行以下操作的數(shù)據(jù)結(jié)構(gòu)。

insert(val)：當元素 val 不存在時，向集合中插入該項。
remove(val)：元素 val 存在時，從集合中移除該項。
getRandom：隨機返回現(xiàn)有集合中的一項。每個元素應(yīng)該有相同的概率被返回。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/insert-delete-getrandom-o1
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請注明出處。

2. 解題

• 哈希表 key存儲數(shù)據(jù)，value存儲其在vector中的下標
• 刪除時，先把vector尾部的數(shù)存到要刪除的位置

class RandomizedSet {vector<int> v;unordered_map<int,int> m; public:RandomizedSet() { }bool insert(int val) {if(m.find(val) == m.end()){m[val] = v.size();v.push_back(val);return true;}elsereturn false;}bool remove(int val) {if(m.find(val) == m.end())return false;v[m[val]] = v.back();m[v.back()] = m[val];v.pop_back();m.erase(val);return true;}int getRandom() {if(v.size() == 0)return -1;return v[rand() % v.size()];} };

總結(jié)

以上是生活随笔為你收集整理的LeetCode 380. 常数时间插入、删除和获取随机元素（哈希+vector）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。