题目描述
运用你所掌握的数据结构,设计和实现一个 LRU (最近最少使用) 缓存机制。它应该支持以下操作: 获取数据 get 和 写入数据 put 。
获取数据 get(key) - 如果关键字 (key) 存在于缓存中,则获取关键字的值(总是正数),否则返回 -1。
写入数据 put(key, value) - 如果关键字已经存在,则变更其数据值;如果关键字不存在,则插入该组「关键字/值」。当缓存容量达到上限时,它应该在写入新数据之前删除最久未使用的数据值,从而为新的数据值留出空间。
进阶:
你是否可以在 O(1) 时间复杂度内完成这两种操作?
分析
LRU是一种常见的页面置换算法,淘汰最近最久没有使用的页面。本题要求我们在O(1)的时间复杂度内实现 get 和 put两种操作。
get操作是获取某一个key对应的value,是一个查询操作,可以想到hash表能够在常数复杂度内实现查询;
put操作是插入一对(key, value),如果key已经存在,就更新对应的value,如果没有就插入。要注意put操作后,key对应的缓存已经变成最新使用的了,此时如果超出了容量限制,就要淘汰最久未使用的,所以put操作包含 更新/插入、调整顺序、超出容量淘汰三个部分。
首先我们知道,如果只使用一个哈希表,能够做到查询、插入、删除是常数时间,但是由于没有顺序,我们不知道淘汰哪一个,因此还需要一个数据结构记录缓存被使用的顺序。如果这一结构采用顺序存储比如vector,那在调整顺序的时候就会出现一部分存储空间整体移动的情况,这不符合时间复杂度要求,因此考虑采用链表的形式存储。
所以,首先我们设计两个数据结构如下:
list<pair<int,int>>l;
unordered_map<int,list<pair<int,int>>::iterator>m;
这里list每个节点保存了一个pair,里面分别是key和value,list的从头到尾的顺序表示从最新使用缓存到最久未用的缓存顺序,也就是说,最新使用一个缓存,就放在l的begin()位置,要淘汰就从末尾开始淘汰。unordered_map则保存key和对应的节点在list中的迭代器(指针)。
在get操作时,通过哈希表find操作查找key能够得到一个unordered_map的迭代器it,那么如果it是m.end()表示没找到,也就是这个key不存在,根据题意返回-1。反之如果存在,那么it->second就是对应缓存保存在list的迭代器,其中it->second->first是key,it->second->second是value,返回value即可。要注意由于key对应的缓存被访问,他应该变成最新的了,应该放在l.begin()的位置,这里用list::splice方法在常数时间内实现(实际上就是调整链表节点指针顺序,不涉及内存拷贝)。显然在常数时间复杂度内能够实现get操作。
put操作,我们首先还是查找m中有没有key,有的话就找到list对应节点删掉,同样的,链表删除只是调整指针并释放内存,是常数时间。然后在list的开始位置,重新插入{key,value},然后更新哈希表记录的key对应的list迭代器信息。这样插入操作就完成了,并且最新插入的也放在了起始位置,然后就是检查是否超出缓存需要淘汰的问题。在淘汰时,我们是从链表最后开始删除,这里使用了反向迭代器rbegin(),找到最后一个节点对应的key并从哈希表移除,将最后一个节点弹出。
stl容器通过迭代器的遍历,常用的正向的方法,c.begin()表示指向第一个元素的迭代器,c.end()表示指向最后一个元素后面一个位置的迭代器 ;类似的反向迭代器 c.rbegin()表示指向最后一个元素的迭代器,而 c.rend()表示指向第一个元素前一个位置的迭代器。
C++代码如下
class LRUCache {
public:
LRUCache(int capacity) {
cap=capacity;
}
int get(int key) {
auto it = m.find(key);
if (it == m.end()) return -1;
l.splice(l.begin(), l, it->second);//put it->second at l.begin()
return it->second->second;
}
void put(int key, int value) {
auto it = m.find(key);
if (it != m.end()) l.erase(it->second);
l.push_front({ key,value });
m[key] = l.begin();
if (m.size() > cap) {
int k = l.rbegin()->first;//反向迭代器
l.pop_back();
m.erase(k);
}
}
list<pair<int,int>>l;
unordered_map<int,list<pair<int,int>>::iterator>m;
int cap=0;
};