1. 背景介绍
LRU(Least Recently Used)缓存是一种常用的缓存淘汰策略,其核心思想是:当缓存空间已满时,优先淘汰最近最少使用的数据。本文将深入探讨LRU缓存的两种不同实现方案,并分析其设计思路和优化技巧。
2. 问题定义
LRU缓存需要支持以下两种基本操作:
get(key)
:如果缓存中存在指定的键,则返回对应的值,并将该数据移动到最近使用的位置。put(key, value)
:将数据写入缓存。如果缓存已满,则需要淘汰最近最少使用的数据。
3. 方案一:自定义双向链表实现
3.1 设计思路
第一个实现方案通过自定义LinkNode
结构和双向链表来管理缓存数据。主要特点包括:
- 使用
unordered_map
存储键值对 - 自定义链表节点,维护数据访问顺序
- 手动管理节点的插入、删除和移动
3.2 核心代码解析
class LRUCache {
public:
struct LinkNode {
int val;
LinkNode* next;
LinkNode* last;
LinkNode() {
val = 0;
next = nullptr;
last = nullptr;
}
};
// 关键实现:get和put方法
int get(int key) {
// 检查key是否存在
if (LRU_map.count(key)) {
// 将访问的节点移动到链表尾部(最近使用)
// 处理不同位置的节点移动情况
...
}
return -1;
}
void put(int key, int value) {
// 处理缓存满、初次插入等多种场景
if (LRU_map.size() == capacity) {
// 淘汰最少使用的数据
...
}
...
}
};
3.3 方案优点与不足
优点:
- 完全自定义实现,对底层细节有更多控制
- 可以精细化内存管理
不足:
- 代码复杂度高
- 需要手动管理内存,容易出错
- 使用多个哈希表,空间利用率不高
4. 方案二:STL优化实现
4.1 设计思路
第二个实现方案利用STL的list
和unordered_map
,显著简化了代码:
- 使用
list<pair<int, int>>
存储键值对 - 使用
unordered_map
维护键与迭代器的映射 - 头部存储最新数据,尾部存储最旧数据
4.2 核心代码解析
class LRUCache {
public:
list<pair<int, int>> cache;
unordered_map<int, list<pair<int, int>>::iterator> hash;
int get(int key) {
if (hash.count(key)) {
// 将访问节点移动到头部
cache.push_front(*hash[key]);
cache.erase(hash[key]);
hash[key] = cache.begin();
return cache.front().second;
}
return -1;
}
void put(int key, int value) {
if (hash.count(key)) {
cache.erase(hash[key]);
}
else if (hash.size() == capacity) {
// 删除最久未使用的数据
hash.erase(cache.back().first);
cache.pop_back();
}
cache.push_front({key, value});
hash[key] = cache.begin();
}
};
4.3 方案优点
- 代码简洁清晰
- 利用STL容器,内存管理更安全
- 减少哈希表数量
- 迭代器操作更高效
5. 性能与空间复杂度分析
操作 | 时间复杂度 | 空间复杂度 |
---|---|---|
get | O(1) | O(capacity) |
put | O(1) | O(capacity) |
6. 总结与建议
- LRU缓存实现可以通过自定义链表或STL容器完成
- 推荐使用STL方案,代码更简洁、更不易出错
- 关键在于维护数据的访问顺序和快速查找
7. 扩展思考
- 如何支持线程安全的LRU缓存?
- 如何实现带过期时间的LRU缓存?
参考链接
注:本文基于LeetCode题解,结合作者原有代码和个人理解撰写。