LRU 介绍
LRU 是 Least Recently Used 的缩写,即最近最少使用,是一种常用的页面置换算法,选择最近最久未使用的页面予以淘汰。
面试中常考的题目,运用你所掌握的数据结构,自己设计手写 LRU 算法。
相关题目可参考 LeetCode146 题
一。使用 LinkedHashMap 实现
在 Java 中,其实 LinkedHashMap 已经实现了 LRU 缓存淘汰算法,需要在构造函数第三个参数传入 true,表示按照时间顺序访问。可以直接继承 LinkedHashMap 来实现。
但是 LinkedHashMap 会自动扩容,如果想实现限制容量删除队列顶端元素,需要重写 removeEldestEntry () 方法,当 map 里面的元素个数大于了缓存最大容量,删除链表的顶端元素。
public class LRULinkedHashMap
private int capacity;
LRULinkedHashMap(int capacity) {
// 初始大小,0.75是装载因子,true是表示按照访问时间排序
super(capacity, 0.75f, true);
//传入指定的缓存最大容量
this.capacity = capacity;
}
/**
* 实现LRU的关键方法,如果map里面的元素个数大于了缓存最大容量,则删除链表的顶端元素
*/
@Override
protected boolean removeEldestEntry(Map.Entry<K, V> eldest) {
return size() > capacity;
}
}
二。使用 LinkedList+HashMap 实现
也可以使用 LinkedList 和 HashMap 实现,但时间复杂度较高。使用 HashMap 可以通过 O (1) 时间拿到元素,但是无法在 O (1) 时间定位它在链表中的位置,在 LinkedList 里访问元素仍然是顺序遍历,所以删除元素的时间复杂度仍然是 O (n)。并不是高效的 Lru 算法。
因为从 HashMap 中删除元素需要 Key,所以这里在链表中存放 Key 而不是 Value。
public class LRUCacheBeta
int capacity;
Map
LinkedList list;
public LRUCacheBeta(int capacity) {
this.capacity = capacity;
this.map = new HashMap<>();
this.list = new LinkedList<>();
}
/**
* 添加元素
* 1.元素存在,放到队尾
* 2.不存在,判断链表是否满。
* 如果满,则删除队首元素,放入队尾元素,删除更新哈希表
* 如果不满,放入队尾元素,更新哈希表
*/
public void put(K key, V value) {
V v = map.get(key);
if (v != null) {
list.remove(key);
list.addLast(key);
map.put(key, value);
return;
}
//队列未满,添加到尾部
if (list.size() < capacity) {
list.addLast(key);
map.put(key, value);
} else {
//队列已满,移除队首
K firstKey = list.removeFirst();
map.remove(firstKey);
list.addLast(key);
map.put(key, value);
}
}
/**
* 访问元素
* 元素存在,放到队尾
*/
public V get(K key) {
V v = map.get(key);
if (v != null) {
list.remove(key);
list.addLast(key);
return v;
}
return null;
}
}
三。使用双向链表结构 + HashMap 实现
在方法二中,删除操作的时间复杂度仍是 O (n),那么如何使其复杂度降为 O (1)?我们可以自定义双向链表的结构,这里定义了内部类 Node,存放 KV 以及前后指针。这样我们通过 hashmap 找到对应 Node,然后根据其前驱节点进行指针的操作,就可以实现复杂度 O (1) 的删除操作。
同样因为访问 HashMap 需要 key,所以定义 Node 节点存放了 K 和 V,而不是只存放 V。保存队列的头节点和尾节点。
在代码中,我们通过调整指针,定义了三个方法,分别是添加元素到队尾,将队列中元素移动到队尾,删除队列头节点并返回,因为是双向链表,特别注意指针变换的顺序以及不要遗漏前驱和后继指针。
public class LRUCache
private int size;
private HashMap
private Node head;
private Node tail;
LRUCache(int size) {
this.size = size;
map = new HashMap<>();
}
/**
* 添加元素
* 1.元素存在,将元素移动到队尾
* 2.不存在,判断链表是否满。
* 如果满,则删除队首元素,放入队尾元素,删除更新哈希表
* 如果不满,放入队尾元素,更新哈希表
*/
public void put(K key, V value) {
Node node = map.get(key);
if (node != null) {
//更新值
node.v = value;
moveNodeToTail(node);
} else {
Node newNode = new Node(key, value);
//链表满,需要删除首节点
if (map.size() == size) {
Node delHead = removeHead();
map.remove(delHead.k);
}
作者:icecrea
链接:https://leetcode-cn.com/problems/lru-cache/solution/san-chong-fang-fa-dai-ni-shou-si-lrusuan-fa-javaba/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。