方法1:利用C++自带的双向链表list和散列表unordered_map
class LRUCache {
public:
LRUCache(int capacity) : cap(capacity) { //构造函数
}
int get(int key) {
if (map.find(key) == map.end()) return -1; //找不到,返回-1 (使用哈希表提高查找速率)
auto key_value = *map[key]; //找到了,根据键得到键值对 //这里必须保存一下,不然下面在cache中erase之后就找不到了
cache.erase(map[key]); //上面找到的键值对是双向链表的节点,直接在链表中删除该节点
cache.push_front(key_value); //将该节点放到链表头
map[key] = cache.begin(); //对哈希表重新进行映射
return key_value.second; //返回
}
void put(int key, int value) {
if (map.find(key) == map.end()) { //找不到,要将该节点放到链表头
if (cache.size() == cap) { //如果链表满了,要删除最后一个节点
map.erase(cache.back().first); //在哈希表中删除映射
cache.pop_back(); //在链表中删除最后一个
}
}
else { //找到了
cache.erase(map[key]); //将原来的删除
}
cache.push_front({key, value}); //将新的放到链表头
map[key] = cache.begin(); //在哈希表中建立映射
}
private:
int cap; //缓存的容量
list<pair<int, int>> cache; //用双向链表实现的缓存
unordered_map<int, list<pair<int, int>>::iterator> map; //哈希表 根据键快速找到值
};
/**
* Your LRUCache object will be instantiated and called as such:
* LRUCache* obj = new LRUCache(capacity);
* int param_1 = obj->get(key);
* obj->put(key,value);
*/
方法2:自己实现双向链表,C++自带的unordered_map
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;
}
};