题解
使用双向链表+hashmap存储数据,链表存储(key, value)和相对位置,hashmap存储(key, 链表节点指针)。
put使用hashmap查找key是否出现并得到元素所在节点,删去原有链表节点并在链表头部插入一个新节点,如果容量大于限制则从链表尾部删除一个元素。
get使用hashmap查找key是否出现并且删除原有节点重新插入在链表头部,实现刷新访问时间功能。
代码使用C++STL,如果要求速度可以手写链表和hashmap。
AC代码
class LRUCache {
public:
int cap;
list<pair<int, int>> ls;
unordered_map<int, list<pair<int, int>>::iterator> mp;
LRUCache(int capacity) {
cap = capacity;
}
int get(int key) {
auto mit = mp.find(key);
if (mit != mp.end())
{
auto lit = mit->second;
int val = lit->second;
ls.erase(lit);
ls.push_front({ key, val });
mit->second = ls.begin();
return val;
}
return -1;
}
void put(int key, int value) {
auto mit = mp.find(key);
if (mit != mp.end())
ls.erase(mit->second);
ls.push_front({ key, value });
mp[key] = ls.begin();
if (ls.size() > cap)
{
key = ls.back().first;
ls.pop_back();
mp.erase(key);
}
}
};