力扣146题:LRU缓存
题目描述
请你设计并实现一个满足 LRU (最近最少使用) 缓存 约束的数据结构。
实现 LRUCache 类:
LRUCache(int capacity) 以 正整数 作为容量 capacity 初始化 LRU 缓存
int get(int key) 如果关键字 key 存在于缓存中,则返回关键字的值,否则返回 -1 。
void put(int key, int value) 如果关键字 key 已经存在,则变更其数据值 value ;如果不存在,则向缓存中插入该组 key-value 。如果插入操作导致关键字数量超过 capacity ,则应该 逐出 最久未使用的关键字。
函数 get 和 put 必须以 O(1) 的平均时间复杂度运行。
输入输出样例
输入
["LRUCache", "put", "put", "get", "put", "get", "put", "get", "get", "get"]
[[2], [1, 1], [2, 2], [1], [3, 3], [2], [4, 4], [1], [3], [4]]
输出
[null, null, null, 1, null, -1, null, -1, 3, 4]
解释
LRUCache lRUCache = new LRUCache(2);
lRUCache.put(1, 1); // 缓存是 {1=1}
lRUCache.put(2, 2); // 缓存是 {1=1, 2=2}
lRUCache.get(1); // 返回 1
lRUCache.put(3, 3); // 该操作会使得关键字 2 作废,缓存是 {1=1, 3=3}
lRUCache.get(2); // 返回 -1 (未找到)
lRUCache.put(4, 4); // 该操作会使得关键字 1 作废,缓存是 {4=4, 3=3}
lRUCache.get(1); // 返回 -1 (未找到)
lRUCache.get(3); // 返回 3
lRUCache.get(4); // 返回 4
解法一,自定义双链表
//自定义双链表
struct DLinkedNode{
//存放键值
int key;
int value;
//定义头指针和尾指针
DLinkedNode *prev;
DLinkedNode *next;
//定义初始化函数
DLinkedNode():key(0),value(0),prev(nullptr),next(nullptr){}
DLinkedNode(int keys,int values):key(keys),value(values),prev(nullptr),next(nullptr){}
};
class LRUCache{
//public 存函数
//private 存放变量
private:
//建立hash表,表中存放 键key 和结点 ,通过对键的查找可以查找到对应的结点
unordered_map<int,DLinkedNode*>cache;
//定义头节点和尾结点
DLinkedNode *head;
DLinkedNode *tail;
int size;
int capacity;
public:
//一旦需要 key 和 value 肯定是 hash表
//要求:1.需要能随机访问 2.能够把数据插入头部或尾部
//使用hash表跟双向链表维护所有在缓冲中的键值对
//双向链表,按照顺序存储键值对,头部是最近使用,尾部是最久未使用
//哈希表,缓存数据键映射到链表中的位置
//构造函数
//初始化变量
LRUCache(int capa)
{
capacity=capa;
size=0;
//构建双链表结点的头指针和尾指针的指向关系
head=new DLinkedNode();
tail=new DLinkedNode();
head->next=tail;
tail->prev=head;
}
//显示数据
//首先判断数据是否存在,通过hash查找结点在双向链表中的位置,并移动到链表头部
int get(int key)
{
//如果键key在hash表中不存在则返回-1
if(!cache.count(key))
{
// cout<<-1<<endl;
return -1;
}
//若键key存在,则通过hash表进行定位,并将其移动到hash表的头部
DLinkedNode *node=cache[key];
//将结点移动到头节点
moveToHead(node);
// cout<<node->value<<endl;
return node->value;
}
//添加函数
//首先判断数据是否存在,若不存在,则新建一个结点,添加到双向链表中和哈希表中,并移动到首部
// 若存在,则先根据hash表进行定位,再将值更新为value,移动到链表的头部
void put(int key,int value)
{
//如果hash表的key不存在,则创建一个新的结点
if(!cache.count(key))
{
DLinkedNode *node=new DLinkedNode(key,value);
//将其值添加到hash表
cache[key]=node;
//并将结点添加到双向链表的头部
addToHead(node);
++size;
//如果hash表中数据的大小大于容量,则删除双向链表的最末尾的结点
if(size>capacity)
{
DLinkedNode *removed=removeTail();
//删除待删除的结点在hash表中对应的项
cache.erase(removed->key);
//删除结点 避免内存泄漏
delete removed;
--size;
}
}
//如果key在hash表中存在,首先通过hash表的定位,再修改value,并且移动到头部
else{
DLinkedNode*node=cache[key];
node->value=value;
moveToHead(node);
}
}
//将结点从当前位置挪出去
void removeNode(DLinkedNode *node)
{
node->prev->next=node->next;
node->next->prev=node->prev;
}
void addToHead(DLinkedNode *node)
{
node->prev=head;
node->next=head->next;
head->next->prev=node;
head->next=node;
}
//删除双向链表的尾结点
DLinkedNode * removeTail()
{
DLinkedNode *node=tail->prev;
removeNode(node);
return node;
}
//首先需要将当前结点从当前位置移除出去
//在将当前结点添加到链表的头部
void moveToHead(DLinkedNode *node)
{
removeNode(node);
addToHead(node);
}
};
解法二:使用链表的迭代器
//使用简便版本的LRUCache
class LRUCache2
{
private:
int capacity;
list<pair<int,int>>cache;
//定义hash表存储键和链表
unordered_map<int,list<pair<int,int>>::iterator>map;
public:
LRUCache2(int cap)
{
capacity=cap;
}
int get(int key)
{
if(!map.count(key))
{
cout<<-1<<endl;
return -1;
}
//key指向的结点
auto key_value=*map[key];
//找到指定的结点,删除当前结点,并将结点插入到链表的头部
cache.erase(map[key]);
cache.push_front(key_value);
map[key]=cache.begin();
//返回当前结点的值
cout<<key_value.second<<endl;
return key_value.second;
}
void put(int key,int value)
{
//如果hash表中不包含key
if(!map.count(key))
{
//hash中的长度刚好等于capacity长度
//将最久的删除
if(cache.size()==capacity)
{
map.erase(cache.back().first);
cache.pop_back();
}
}
//如果存在则将当前的结点先删除
else{
cache.erase(map[key]);
}
//插入头节点
cache.push_front({key,value});
map[key]=cache.begin();
}
};