前言
提示:在CMU15-445的lab2中我们需要实现一个数据库page的控制机制,也就是LRU缓存机制,即当内存容量达到上限时,我们应该移除最久未使用的page。
一、题目描述?
运用你所掌握的数据结构,设计和实现一个 LRU (最近最少使用) 缓存机制 。
实现 LRUCache 类:
LRUCache(int capacity) 以正整数作为容量 capacity 初始化 LRU 缓存
int get(int key) 如果关键字 key 存在于缓存中,则返回关键字的值,否则返回 -1 。
void put(int key, int value) 如果关键字已经存在,则变更其数据值;如果关键字不存在,则插入该组「关键字-值」。当缓存容量达到上限时,它应该在写入新数据之前删除最久未使用的数据值,从而为新的数据值留出空间。
二、关键点
(1)有key-value我们推断需要哈希表。
(2)需要存储数据的访问时间,定义越靠前则最近刚刚被访问过。
存储思想:
1.存储一个有序序列,若某个key被访问过了,便使其移动到该序列的最前面。
2.因此需要经常移动数据,最好使用链表
(3)将key访问与存储的数据访问时间序列结合起来。即对于链表中存储key值
具体运行过程:
(1)当需插入键值对且内存未满时,将键值对存储到HashMap中,并将一个key的node插入到链表的最前端。
(2)当某个数据被访问后,将该key的node移动到链表的最前方
(3)当需插入键值对且内存已满时,先将链表最后方的node值和HashMap中的对应键值对删除。再将目标键值对存储到HashMap中,并将一个目标key的node插入到链表的最前端。
三、优化
在上文中,我们仍需要迭代的在链表中查找我们的目标key值,在对其进行移动,即无法在访问时间链表中对key值进行随机访问,这极大的提升了复杂度。
那如何可以随机访问链表的key值?
解决方案:
我们可以在 HashMap 中,存储当前 key 在时序列表链表中的节点地址。这样我们需要访问一个key值时,我们可以访问其HashMap的key值,然后得到链表中的key的node地址。即对于在链表中我们便实现了随机访问。
小问题:
但我们在移动节点的时候仍然需要迭代的查找前一节点的node,我们可以使用双向链表来存储访问时间序列
四,代码
public class LRUCache {
class DLinkedNode {
int key;
int value;
DLinkedNode prev;
DLinkedNode next;
public DLinkedNode() {}
public DLinkedNode(int _key, int _value) {key = _key; value = _value;}
}
private Map<Integer, DLinkedNode> cache = new HashMap<Integer, DLinkedNode>();
private int size;
private int capacity;
private DLinkedNode head, tail;
public LRUCache(int capacity) {
this.size = 0;
this.capacity = capacity;
// 使用伪头部和伪尾部节点
head = new DLinkedNode();
tail = new DLinkedNode();
head.next = tail;
tail.prev = head;
}
public int get(int key) {
DLinkedNode node = cache.get(key);
if (node == null) {
return -1;
}
// 如果 key 存在,先通过哈希表定位,再移到头部
moveToHead(node);
return node.value;
}
public void put(int key, int value) {
DLinkedNode node = cache.get(key);
if (node == null) {
// 如果 key 不存在,创建一个新的节点
DLinkedNode newNode = new DLinkedNode(key, value);
// 添加进哈希表
cache.put(key, newNode);
// 添加至双向链表的头部
addToHead(newNode);
++size;
if (size > capacity) {
// 如果超出容量,删除双向链表的尾部节点
DLinkedNode tail = removeTail();
// 删除哈希表中对应的项
cache.remove(tail.key);
--size;
}
}
else {
// 如果 key 存在,先通过哈希表定位,再修改 value,并移到头部
node.value = value;
moveToHead(node);
}
}
private void addToHead(DLinkedNode node) {
node.prev = head;
node.next = head.next;
head.next.prev = node;
head.next = node;
}
private void removeNode(DLinkedNode node) {
node.prev.next = node.next;
node.next.prev = node.prev;
}
private void moveToHead(DLinkedNode node) {
removeNode(node);
addToHead(node);
}
private DLinkedNode removeTail() {
DLinkedNode res = tail.prev;
removeNode(res);
return res;
}
}
小结
将链表与HashMap结合起来后,可以实现对链表的随机访问。