第一次写这种类型的程序,感觉挺好玩的.LRU就是 Least Recently Used.LRU Cache就是一种缓存机制,当缓存已经满了的时候,最近最少使用的元素移出缓存换上新的元素.
我们采用双向链表来判断最近最少使用的元素.我们保存指向双向链表的所有元素的指针,当访问一个元素的时候,把双向链表中的该元素移动到队尾.即只要使用某个元素,就把该元素移动到队尾,所以队尾是最近最多使用的元素,队首即最近使用最少的元素.当插入新元素的时候且cache容量已经到的时候,把队首元素删除就好.
class LRUCache {
public:
LRUCache(int capacity) {
this->_capacity=capacity;
}
int get(int key) {
if(cache.find(key)!=cache.end()){
freq.erase(iter_map[key]);
freq.push_back(key);
list<int>::iterator it=freq.end();
--it;
iter_map[key]=it;
return cache[key];
}else return -1;
}
void put(int key, int value) {
if(cache.find(key)!=cache.end()){
freq.erase(iter_map[key]);
freq.push_back(key);
list<int>::iterator it=freq.end();
--it;
iter_map[key]=it;
cache[key]=value;
}else if((int)cache.size()<_capacity){
cache[key]=value;
freq.push_back(key);
list<int>::iterator it=freq.end();
--it;
iter_map[key]=it;
}else{
int front=freq.front();
iter_map.erase(front);
cache.erase(front);
freq.erase(freq.begin());
cache[key]=value;
freq.push_back(key);
list<int>::iterator it=freq.end();
--it;
iter_map[key]=it;
}
return ;
}
private:
int _capacity;
list<int> freq;
unordered_map<int,int> cache;
unordered_map<int,list<int>::iterator> iter_map;
};
/**
* Your LRUCache object will be instantiated and called as such:
* LRUCache* obj = new LRUCache(capacity);
* int param_1 = obj->get(key);
* obj->put(key,value);
*/