LRU缓存机制——简化版的LinkedHashMap

今天是个综合的题目，实现LRU。原题目地址：leetcode

运用你所掌握的数据结构，设计和实现一个  LRU (最近最少使用) 缓存机制。它应该支持以下操作： 获取数据 get 和 写入数据 put 。

获取数据 get(key) - 如果密钥 (key) 存在于缓存中，则获取密钥的值（总是正数），否则返回 -1。
写入数据 put(key, value) - 如果密钥已经存在，则变更其数据值；如果密钥不存在，则插入该组「密钥/数据值」。当缓存容量达到上限时，它应该在写入新数据之前删除最久未使用的数据值，从而为新的数据值留出空间。

 

进阶:
你是否可以在 O(1) 时间复杂度内完成这两种操作？

 

示例:
LRUCache cache = new LRUCache( 2 /* 缓存容量 */ );

cache.put(1, 1);
cache.put(2, 2);
cache.get(1);       // 返回  1
cache.put(3, 3);    // 该操作会使得密钥 2 作废
cache.get(2);       // 返回 -1 (未找到)
cache.put(4, 4);    // 该操作会使得密钥 1 作废
cache.get(1);       // 返回 -1 (未找到)
cache.get(3);       // 返回  3
cache.get(4);       // 返回  4

这里用现成的类库，就可以实现——LinkedHashMap，这是最省事的的

class LRUCache extends LinkedHashMap<Integer, Integer>{
    private int capacity;
    
    public LRUCache(int capacity) {
        super(capacity, 0.75F, true);
        this.capacity = capacity;
    }

    public int get(int key) {
        return super.getOrDefault(key, -1);
    }

    public void put(int key, int value) {
        super.put(key, value);
    }

    @Override
    protected boolean removeEldestEntry(Map.Entry<Integer, Integer> eldest) {
        return size() > capacity; 
    }
}

若自己实现，并且两个操作的时间复杂度为O(1)。一个双向链表，外加一个哈希函数就可以了

class LRUCache {
    class DLindNode{
        int key;
        int value;
        DLindNode pre;
        DLindNode next;
        public DLindNode(){}
        public DLindNode(int _key, int _value){
            key = _key;
            value = _value;
        }
    }
    int size;
    int capacity;
    DLindNode head,tail;
    Map<Integer, DLindNode> cache = new HashMap<Integer,DLindNode>();
    public LRUCache(int capacity) {
        this.size = 0;
        this.capacity = capacity;
        head = new DLindNode();
        tail = new DLindNode();
        head.next = tail;
        tail.pre = head;
    }
    
    public int get(int key) {
        DLindNode node = cache.get(key);
        if(null == node){
            return -1;
        }
        moveToHead(node);
        return node.value;
    }
    
    public void put(int key, int value) {
        DLindNode node = cache.get(key);
        if(node != null){
            node.value = value;
            moveToHead(node);
            return;
        }
        node = new DLindNode(key, value);
        if(size >= capacity){
            DLindNode removeNode = removeTail();
            cache.remove(removeNode.key);
            size--;
        }
        size++;
        addToHead(node);
        cache.put(key, node);
    }
    private DLindNode removeTail(){
        DLindNode node = tail.pre;
        node.pre.next = tail;
        tail.pre = node.pre;
        return node;
    }

    private void moveToHead(DLindNode node){
        removeNode(node);
        addToHead(node);
    }

    private void addToHead(DLindNode node){
        node.next = head.next;
        node.pre = head;
        head.next.pre = node;
        head.next = node;
    }
    private void removeNode(DLindNode node){
        node.pre.next = node.next;
        node.next.pre = node.pre;
    }
}

代码不复杂，就不过多解释了。