LRU缓存
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> map = new HashMap<>();
private int size;
private int capacity;
private DLinkedNode head,tail;
public LRUCache(int capacity) {
this.size = 0;
this.capacity = 2;
head = new DLinkedNode();
tail = new DLinkedNode();
head.next = tail;
tail.next = head;
}
public int get(int key) {
DLinkedNode node = map.get(key);
if (node == null){
return -1;
}
moveToHead(node);
return node.value;
}
public void put (int key,int value){
DLinkedNode node = map.get(key);
if (node == null){
DLinkedNode newNode = new DLinkedNode(key,value);
map.put(key,newNode);
addToHead(newNode);
++size;
if (size>capacity){
DLinkedNode res = removeTail();
map.remove(res.key);
size--;
}else {
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.prev.next = node.prev;
}
private void moveToHead(DLinkedNode node) {
removeNode(node);
addToHead(node);
}
private DLinkedNode removeTail() {
DLinkedNode res = tail.prev;
removeNode(res);
return res;
}
}