方法一:哈希表 + 双向链表(官方)
我们用一个哈希表和一个双向链表维护所有在缓存中的键值对。我们用一个哈希表和一个双向链表维护所有在缓存中的键值对;我们用一个哈希表和一个双向链表维护所有在缓存中的键值对。我们首先使用哈希表进行定位,找出缓存项在双向链表中的位置,随后将其移动到双向链表的头部,即可在 O(1)的时间内完成 get 或者 put 操作。
- 对于 get 操作,首先判断 key 是否存在:如果 key 不存在,则返回 −1;如果 key 存在,则 key 对应的节点是最近被使用的节点。通过哈希表定位到该节点在双向链表中的位置,并将其移动到双向链表的头部,最后返回该节点的值。
- 对于 put 操作,首先判断 key 是否存在:如果 key 不存在,使用 key 和 value 创建一个新的节点,在双向链表的头部添加该节点,并将 key 和该节点添加进哈希表中。然后判断双向链表的节点数是否超出容量,如果超出容量,则删除双向链表的尾部节点,并删除哈希表中对应的项;如果 key 存在,则与 get 操作类似,先通过哈希表定位,再将对应的节点的值更新为 value,并将该节点移到双向链表的头部。
struct DLinkedNode {
int key, value;
DLinkedNode* prev;
DLinkedNode* next;
DLinkedNode(): key(0), value(0), prev(nullptr), next(nullptr) {}
DLinkedNode(int _key, int _value): key(_key), value(_value), prev(nullptr),
next(nullptr) {}
};
class LRUCache {
private:
unordered_map<int, DLinkedNode*> cache;
DLinkedNode* head;
DLinkedNode* tail;
int size;
int capacity;
public:
LRUCache(int _capacity): capacity(_capacity), size(0) {
// 使用伪头部和伪尾部节点
head = new DLinkedNode();
tail = new DLinkedNode();
head->next = tail;
tail->prev = head;
}
int get(int key) {
if (!cache.count(key)) {
return -1;
}
// 如果 key 存在,先通过哈希表定位,再移到头部
DLinkedNode* node = cache[key];
moveToHead(node);
return node->value;
}
void put(int key, int value) {
if (!cache.count(key)) {
// 如果 key 不存在,创建一个新的节点
DLinkedNode* node = new DLinkedNode(key, value);
// 添加进哈希表
cache[key] = node;
// 添加至双向链表的头部
addToHead(node);
++size;
if (size > capacity) {
// 如果超出容量,删除双向链表的尾部节点
DLinkedNode* removed = removeTail();
// 删除哈希表中对应的项
cache.erase(removed->key);
// 防止内存泄漏
delete removed;
--size;
}
}
else {
// 如果 key 存在,先通过哈希表定位,再修改 value,并移到头部
DLinkedNode* node = cache[key];
node->value = value;
moveToHead(node);
}
}
void addToHead(DLinkedNode* node) {
node->prev = head;
node->next = head->next;
head->next->prev = node;
head->next = node;
}
void removeNode(DLinkedNode* node) {
node->prev->next = node->next;
node->next->prev = node->prev;
}
void moveToHead(DLinkedNode* node) {
removeNode(node);
addToHead(node);
}
DLinkedNode* removeTail() {
DLinkedNode* node = tail->prev;
removeNode(node);
return node;
}
};
使用双向链表而不是单向链表的原因是:在缓存满了需要删除的时候,双相链表可以通过tail快速找到最后一个节点。如果使用单向链表想要找到最后一个节点,需要从头节点遍历过去。
方法二、list+hashmap
双链表存储一个节点被使用(get或者put)的时间戳,且按最近使用时间从左到右排好序,最先被使用的节点放在双链表的第一位,因此双链表的最后一位就是最久未被使用的节点;
get(key):
首先用哈希表判断key是否存在:如果key不存在,则返回-1;如果key存在,则返回对应的value,同时将key对应的节点放到双链表的最左侧;
put(key, value):
首先用哈希表判断key是否存在:如果key存在,则修改对应的value,同时将key对应的节点放到双链表的最左侧;如果key不存在:如果缓存已满,则删除双链表最右侧的节点(上次使用时间最老的节点),更新哈希表;否则,插入(key, value):同时将key对应的节点放到双链表的最左侧,更新哈希表;
class LRUCache {
public:
struct Node{
int key,value;
Node*left,*right;
Node(int _key,int _value):key(_key),value(_value),left(NULL),right(NULL){}
}*L,*R;
int n;
unordered_map<int,Node*>hash;
void remove(Node*p){
p->right->left=p->left;
p->left->right=p->right;
}
void insert(Node*p){
p->right=L->right;
p->left=L;
L->right->left=p;
L->right=p;
}
LRUCache(int capacity) {
n=capacity;
L=new Node(-1,-1),R=new Node(-1,-1);
L->right=R;
R->left=L;
}
int get(int key) {
if(hash.count(key)==0)return -1;
auto p=hash[key];
remove(p);
insert(p);
return p->value;
}
void put(int key, int value) {
if(hash.count(key)){
auto p=hash[key];
p->value=value;
remove(p);
insert(p);
}
else{
if(hash.size()==n){
auto p=R->left;
remove(p);
hash.erase(p->key);
delete p;
}
auto p=new Node(key,value);
hash[key]=p;
insert(p);
}
}
};