LeetCode-146.LRU缓存机制
实现 LRUCache 类:
LRUCache(int capacity) 以正整数作为容量 capacity 初始化 LRU 缓存
int get(int key) 如果关键字 key 存在于缓存中,则返回关键字的值,否则返回 -1 。
void put(int key, int value) 如果关键字已经存在,则变更其数据值;如果关键字不存在,则插入该组「关键字-值」。当缓存容量达到上限时,它应该在写入新数据之前删除最久未使用的数据值,从而为新的数据值留出空间。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lru-cache
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
LRU(操作系统页面替换算法):
遵循最近最少被使用原则,每次移除最近最少被使用的界面,依据此原则,每次节点被访问或新创建的结点就移到链表首部
主要思路:哈希表快速访问 + 双链表(快速增删)+
双链表结构实现:
typedef struct DoubleLinkNode{
int key;
int val;
struct DoubleLinkNode *next;
struct DoubleLinkNode *pre;
DoubleLinkNode():key(0),val(0),next(nullptr),pre(nullptr){}
DoubleLinkNode(int _key,int _val):key(_key),val(_val),next(nullptr),pre(nullptr){}
}DLinkNode;
哈希表采用 unordered_map,其是C++11新增架构包含在头文件<unordered_map>中
unordered_map与map的区别与联系:
unordered_map内部实现哈希表,所以查找速度很快
map内存使用红黑树实现,增删改查都可在o(logn)实现
两则都提供了insert,size,count等操作
LRU缓存机制中get思路:
//哈希快速得到缓存内是否存在该键,存在的话则直接移到头部,以使满足LRU机制
int get(int key) {
if(!cache.count(key)){
return -1;
}
DLinkNode *node = cache[key];
moveHead(node);
return node->val;
}
put实现思路:
节点以存在则修改其值就行,因被访问了,所以移到链表首部
不存在时:新建结点,加入哈希表和链表,移到链表首部
容量超时:删除链尾部节点(注意delete防止内存泄露)
void put(int key, int value) {
//不存在这个结点,新建结点
if(!cache.count(key)){
DLinkNode *node = new DLinkNode(key,value);
cache[key] = node;
addHead(node);
++size;
if(size > capacity){//删除结点时一定要delete才能完全删除,避免内存泄露
DLinkNode *removed = tail->pre;
removeNode(removed);
cache.erase(removed->key);
delete removed;
--size;
}
}else{
DLinkNode *node = cache[key];
node->val = value;
moveHead(node);
}
}
具体实现代码:
typedef struct DoubleLinkNode{
int key;
int val;
struct DoubleLinkNode *next;
struct DoubleLinkNode *pre;
DoubleLinkNode():key(0),val(0),next(nullptr),pre(nullptr){}
DoubleLinkNode(int _key,int _val):key(_key),val(_val),next(nullptr),pre(nullptr){}
}DLinkNode;
class LRUCache {
unordered_map<int,DLinkNode*> cache;
DLinkNode *head;
DLinkNode *tail;
int size;
int capacity;
public:
//初始化链表容量,创建头尾结点,提高效率
LRUCache(int capacity) :capacity(capacity),size(0){
head = new DLinkNode();
tail = new DLinkNode();
head->next = tail;
tail->pre = head;
}
//访问了这个结点,移到链表头部
int get(int key) {
if(!cache.count(key)){
return -1;
}
DLinkNode *node = cache[key];
moveHead(node);
return node->val;
}
void put(int key, int value) {
//不存在这个结点,新建结点
if(!cache.count(key)){
DLinkNode *node = new DLinkNode(key,value);
cache[key] = node;
addHead(node);
++size;
if(size > capacity){//删除结点时一定要delete才能完全删除,避免内存泄露
DLinkNode *removed = tail->pre;
removeNode(removed);
cache.erase(removed->key);
delete removed;
--size;
}
}else{
DLinkNode *node = cache[key];
node->val = value;
moveHead(node);
}
}
void addHead(DLinkNode *node){
node->next = head->next;
head->next->pre = node;
node->pre = head;
head->next = node;
}
void removeNode(DLinkNode *node){
node->pre->next = node->next;
node->next->pre = node->pre;
}
void moveHead(DLinkNode *node){
removeNode(node);
addHead(node);
}
};