众所周知,数组中插入,删除元素时间最坏情况下时间复杂度O(n),这是由于,当一个位置的元素发生改变时,其他元素都会受到影响,那么有无简单方式实现常数级时间复杂度实现插删呢?
有的兄弟!那就是哈希 + 动态数组
总的来说,哈希用于定位元素位置,并且作用于数组中,这个过程,数组尾部元素立大功
具体算法过程如下(以删除为例):
1.unorder<int, int>hash中key表示元素值,value表示元素索引(正好相反hhh)利用hash特性可以迅速获取元素下标,然后,把将要删除的元素的下标给到hash表末尾元素,相当于以后查找hash表,末尾元素都会索引到被删除的元素的位置,最后将hash中的元素记录erase就好了
2.对于数组部分的处理,我们用数组末尾元素覆盖待删除的元素(下标步骤1就得到了),然后直接pop()弹出末尾元素就好了。
下面是代码:
//常数时间复杂度实现数组中插入,删除元素
class RandomizedSet {
public:
RandomizedSet() {
}
bool insert(int val) {
if(hash.find(val) != hash.end())return false;
hash[val] = arr.size();
arr.push_back(val);
return true;
}
bool remove(int val) {
if(hash.find(val) == hash.end())return false;
int idx = hash[val];
hash[arr.back()] = idx; //将最后一个hash元素索引到将要删除的数组的位置
hash.erase(val);
arr[idx] = arr.back();
arr.pop_back();
return true;
}
int getRandom() {
return arr[rand() % arr.size()];
}
private:
vector<int> arr;
unordered_map<int, int> hash;
};
/**
* 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();
*/
算法步骤+代码,结合起来看,还是很好理解的。
977

被折叠的 条评论
为什么被折叠?



