运用你所掌握的数据结构,设计和实现一个 LRU (最近最少使用) 缓存机制。它应该支持以下操作: 获取数据 get 和 写入数据 put 。
获取数据 get(key) - 如果密钥 (key) 存在于缓存中,则获取密钥的值(总是正数),否则返回 -1。
写入数据 put(key, value) - 如果密钥不存在,则写入其数据值。当缓存容量达到上限时,它应该在写入新数据之前删除最近最少使用的数据值,从而为新的数据值留出空间。
进阶:
你是否可以在 O(1) 时间复杂度内完成这两种操作?
示例:
LRUCache cache = new LRUCache( 2 /* 缓存容量 */ );
cache.put(1, 1);
cache.put(2, 2);
cache.get(1); // 返回 1
cache.put(3, 3); // 该操作会使得密钥 2 作废
cache.get(2); // 返回 -1 (未找到)
cache.put(4, 4); // 该操作会使得密钥 1 作废
cache.get(1); // 返回 -1 (未找到)
cache.get(3); // 返回 3
cache.get(4); // 返回 4
我的思路
这道题的思路就是用双向链表和map即可实现,STL中链表和hash已经做了封装,所以不用我们去实现。链表中存储key值,而hash中存储key的位置(迭代器)以及value值。为了方便理解,我使用了两个map,由于map中不需要有序,用hashmap即可,空间换时间。
代码
class LRUCache {
public:
int size=0;
unordered_map<int,list<pair<int,int>>::iterator> m;
list<pair<int,int>> l;//key,value
LRUCache(int capacity) {
size=capacity;
}
int get(int key) {
if(m.count(key)){ // 命中
put(key,m[key]->second);// 把当前节点再次放入 LRUCache
return m[key]->second;
}
return -1;
}
void put(int key, int value) {
if (m.count(key)) {
l.erase(m[key]);
m.erase(key); // 可以省略
}
l.push_front({key,value});// 头插
m[key]=l.begin();// 头插所以迭代器是begin
if(m.size()>size) { //长度超过,考虑移除最后的
m.erase(l.back().first);
l.pop_back();
}
}
};