思路:借鉴了网上的讨论,用一个unordered_map和list组合数据结构,unordered_map存储关键字是否存在,list存储相应的值,最近使用的放在头部,否则放在尾部,unordered_map的第二项存储值在list中的位置,即迭代器。然后知道LRU的原理,get(key)需要将key对应的值放在list的头部,m[key]要做相应的变换,这个刚开始没有写上,挂了好多次。后面的set(key,value)反到是容易很多,考虑已经存储的元素大小size和容量capacity的关系,size<capacity时直接插入list,否则删除list末尾的元素再来插入key到list头部,当然,每次更新list一定要记得对map[key]值做相应的变换,以便我们知道key对应的value存在于list的何处。
class LRUCache{
public:
LRUCache(int capacity) {
cap = capacity;
size = 0;
}
int get(int key) {
if(cap<1)return -1;
unordered_map<int,list<int>::iterator>::iterator it;
it = m.find(key);
if(it != m.end()){
list<int>::iterator list_it = it->second;
int ret = *list_it;
L.erase(list_it);
L.push_front(ret);
m[key] = L.begin();
return ret;
}
else
return -1;
}
void set(int key, int value) {
if(cap<1)return;
unordered_map<int,list<int>::iterator>::iterator it;
it = m.find(key);
if(it!=m.end()){
list<int>::iterator list_it = it->second;
L.erase(list_it);
L.push_front(value);
m[key] = L.begin();
}
else{
if(size <cap){
L.push_front(value);
m[key] = L.begin();
size++;
}
else{
list<int>::iterator list_it2 = L.end();
list_it2--;
unordered_map<int,list<int>::iterator>::iterator it2;
for(it2= m.begin();it2!=m.end();it2++)
if(it2->second == list_it2) break;
//cout<<"hello"<<endl;
if(it2!=m.end()){
m.erase(it2);
L.pop_back();
L.push_front(value);
m[key] = L.begin();
}
}
}
}
void output(){
cout<<cap<<" "<<size<<endl;
cout<<*L.begin()<<endl;
}
private:
int cap;
int size;
list<int> L; //存储value
unordered_map<int,list<int>::iterator> m; //存储关键字在list中的位置
};
其中有一个小疑问:在
if(it2!=m.end()){
m.erase(it2);
L.pop_back();
L.push_front(value);
m[key] = L.begin();
}
如果去掉if判断,直接去操作,会出现RE,不知怎么回事,回头再想。