[力扣 Hot100]Day35 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) 的平均时间复杂度运行。
在这里插入图片描述
出处

思路

写了两个版本都超时，单向链表和迭代器的查找都太慢，需要用哈希表。

代码

class LRUCache {
private:int len;list<pair<int, int>> lru;unordered_map<int, list<pair<int, int>>::iterator> map;
public:LRUCache(int capacity) {len=capacity;}int get(int key) {if (map.find(key) == map.end()) return -1;pair<int,int> key_value = *map[key];lru.erase(map[key]);lru.push_front(key_value);map[key] = lru.begin();return key_value.second;}void put(int key, int value) {if (map.find(key) == map.end()) {if (lru.size() == len) {map.erase(lru.back().first);lru.pop_back();}}else {lru.erase(map[key]);}lru.push_front({key, value});map[key] = lru.begin();}
};