146. LRU 缓存机制(Java实现)

要注意的点：双向链表的头尾节点都是伪节点。

class LRUCache {
    // 双向链表
    class DLinkedNode{
        int val;
        int key;
        DLinkedNode next, prev;
        public DLinkedNode(){}
        public DLinkedNode(int key, int val){
            this.key = key;
            this.val = val;
        }
    }

    private int size;
    private int capacity;
    private HashMap<Integer, DLinkedNode> map = new HashMap<>();
    // 伪链表头和链表尾
    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 = map.get(key);
        if(node == null){
            return -1;
        }
        // 如果该节点存在，把它移到链表最前面
        moveToHead(node);
        return node.val;
    }
    
    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 tail = removeTail();
                map.remove(tail.key);
                size--;
            }
        }else{
            // 更新node的值
            node.val = value;
            // 如果该节点存在，把它移到链表最前面
            moveToHead(node);
        }
    }

    // 把节点移到链表最前面
    public void moveToHead(DLinkedNode node){
        // 先从链表中删除
        removeNode(node);
        // 删完之后头插法
        addToHead(node);
    }

    // 头插法
    public void addToHead(DLinkedNode node){
        node.prev = head;
        head.next.prev = node;
        head.next = node;
    }

    public void removeNode(DLinkedNode node){
        node.prev.next = node.next;
        node.next.prev = node.prev;
    }
    // 删除结点还要返回是因为要map中要删除相应的键值对，所以将key返回
    public DLinkedNode removeTail(){
        DLinkedNode res = tail.prev;
        removeNode(res);
        return res;
    }
}

/**
 * Your LRUCache object will be instantiated and called as such:
 * LRUCache obj = new LRUCache(capacity);
 * int param_1 = obj.get(key);
 * obj.put(key,value);
 */