import java.util.HashMap;
class LRUCache {
class Node {
public int key, val;
public Node next, prev;
public Node(int k, int v) {
this.key = k;
this.val = v;
}
}
// key -> Node(key, val)
private HashMap<Integer, Node> map;
// Node(k1, v1) <-> Node(k2, v2)...
private DoubleList cache;
// 最大容量
private int cap;
class DoubleList {
Node head = new Node(0, 0);
Node tail = new Node(0, 0);
int size;
private DoubleList() {
head.next = tail;
tail.prev = head;
size = 0;
}
private void addFirst(Node n) {
Node headNext = head.next;
head.next = n;
headNext.prev = n;
n.prev = head;
n.next = headNext;
size++;
}
private void remove(Node n) {
n.prev.next = n.next;
n.next.prev = n.prev;
size--;
}
private Node removeLast() {
Node last = tail.prev;
remove(last);
return last;
}
private int size() {
return size;
}
}
public LRUCache(int capacity) {
this.cap = capacity;
map = new HashMap<>();
cache = new DoubleList();
}
public int get(int key) {
if (!map.containsKey(key))
return -1;
int val = map.get(key).val;
// 利用 put 方法把该数据提前
put(key, val);
return val;
}
public void put(int key, int value) {
//先把新节点 x 做出来
Node x = new Node(key, value);
if (map.containsKey(key)) {
// 删除旧的节点,新的插到头部
cache.remove(map.get(key));
cache.addFirst(x);
// 更新 map 中对应的数据
map.put(key, x);
} else {
if (cap == cache.size()) {
// 删除链表最后一个数据
Node last = cache.removeLast();
map.remove(last.key);
}
// 直接添加到头部
cache.addFirst(x);
map.put(key, x);
}
}
}
···
LRUCache
最新推荐文章于 2024-05-02 11:11:16 发布