146. LRU缓存机制
运用你所掌握的数据结构,设计和实现一个 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
解题思路
- 哈希表unordererd_map+双向链表实现
//LRU缓存机制
//哈希表unordererd_map+双向链表实现
struct DLinkList {
int key;
int value;
DLinkList* prev;//前驱
DLinkList* next;//后继
DLinkList():key(0),value(0),prev(NULL),next(NULL){}//默认构造函数
DLinkList(int _key, int _value){
key = _key;
value = _value;
}
};
class LRUCache {
private:
unordered_map<int,DLinkList*> mp;//哈希表,存key在链表中的结点
DLinkList* DummyHead;//虚拟头结点
DLinkList* DummyRail;//虚拟尾结点
int curCapacity;//当前容量
int capacity;//总容量
public:
//构造函数
LRUCache(int capacity) {
this->capacity = capacity;
curCapacity = 0;
//初始化双向链表
DummyHead = new DLinkList();
DummyRail = new DLinkList();
DummyHead->next = DummyRail;
DummyRail->prev = DummyHead;
}
int get(int key) {
//判断key是否存在
if(mp[key]){
//存在:将key所在的结点移到链头
DLinkList* tmp = mp[key];
tmp->prev->next = tmp->next;
tmp->next->prev = tmp->prev;
tmp->prev = DummyHead;
tmp->next = DummyHead->next;
DummyHead->next = tmp;
tmp->next->prev = tmp;
return tmp->value;
}
//不存在:返回-1
else
return -1;
}
void put(int key, int value) {
//判断key是否存在
//已存在
if(mp[key]){
//修改对应value值
DLinkList* tmp = mp[key];
tmp->key = key;
tmp->value = value;
//移到链头
tmp->prev->next = tmp->next;
tmp->next->prev = tmp->prev;
tmp->prev = DummyHead;
tmp->next = DummyHead->next;
DummyHead->next = tmp;
tmp->next->prev = tmp;
}
//不存在
else {
//判断容量是否满了
if(curCapacity==capacity){
//删除链尾结点(最近最久未使用)
DLinkList* tmp = DummyRail->prev;
DummyRail->prev = tmp->prev;
DummyRail->prev->next = DummyRail;
//哈希表中也删除该结点*
mp.erase(tmp->key);
//内存中删除该节点*
delete tmp;
//当前容量-1
curCapacity--;
}
//链头插入key
DLinkList* tmp = new DLinkList(key,value);
tmp->next = DummyHead->next;
tmp->prev = DummyHead;
DummyHead->next = tmp;
tmp->next->prev = tmp;
//将key加入哈希表
mp[key] = tmp;
//当前容量+1
curCapacity++;
}
}
};
执行用时:188 ms, 在所有 C++ 提交中击败了87.41% 的用户
内存消耗:37.3 MB, 在所有 C++ 提交中击败了60.12% 的用户