LRU缓存实现(手写版)

运用你所掌握的数据结构,设计和实现一个  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
 
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lru-cache
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
 
实现思路:查找和插入的时间复杂度都需要是O(1),那么只能通过hash表+双向链表的方式实现(LinkedHashMap就是这种方式),这里首先使用自定义的一个双向链表进行实现。
 
 
    private int headKey;
 
    private int lastKey;
 
    private Map<Integer,Node> map;
 
    private int capacity;
 
    public LRUCache(int capacity) {
        headKey = -1;
        lastKey = -1;
        this.capacity = capacity;
        map = new HashMap<>(capacity);
    }
 
    /**
     * 将节点移动到尾节点
     * @param n
     */
    private void changeNodeToLast(Node n){
        //判断是否有头节点
        if(headKey == -1){
            headKey = n.key;
        }
        //判断当前节点是否是尾节点
        if(lastKey == n.key){
            return;
        }
        //判断当前节点是否是头节点
        if(headKey == n.key){
            //判断当前是否只有一个节点
            if(n.after == null){
                return;
            }
            headKey = n.after.key;
        }
        Node b = n.before;
        Node a = n.after;
        //将当前节点移动到尾节点
        if(b != null){
            b.after = a;
        }
        if(a != null){
            a.before = b;
        }
        Node last = map.get(lastKey);
        lastKey = n.key;
        n.before = last;
        last.after = n;
        last = n;
        last.after = null;
    }
 
    /**
     * 添加新节点
     * @param key
     * @return
     */
    private Node addNewNode(int key){
        Node n = new Node();
        //是否有头节点
        if(headKey == -1){
            headKey = key;
        }
        //是否有尾节点
        else if(lastKey == -1){
            lastKey = key;
            Node head = map.get(headKey);
            n.before = head;
            head.after = n;
        }else{
            //中间节点处理
            Node last = map.get(lastKey);
            n.before = last;
            last.after = n;
            lastKey = key;
        }
        return n;
    }
 
    /**
     * 移除头节点
     */
    private void removeNode(){
        Node node = map.get(headKey);
        map.remove(headKey);
        if(node.after != null){
            headKey = node.after.key;
            node.after.before = null;
        }else{
            headKey = -1;
            lastKey = -1;
        }
    }
    
    public int get(int key) {
        Node n = map.get(key);
        if(n != null){
            this.changeNodeToLast(n);
            return n.value;
        }
        return -1;
    }
    
    public void put(int key, int value) {
        Node n = map.get(key);
        if(n == null){
            if(map.size()>= capacity){
               this.removeNode();
            }
            n = this.addNewNode(key);
            n.key = key;
            n.value = value;
            map.put(key,n); 
        }else{ 
           this.changeNodeToLast(n);
           n.key = key;
           n.value = value;
        }
    }
 
    private class Node{
        public Node before;
 
        public int key ;
 
        private int value;
 
        private Node after;
 
        public Node(){
            key = -1;
            value = -1;
        }
 
    }
力扣执行结果:

 

 下一章通过继承LinkedHashMap进行实现。https://www.cnblogs.com/ymqj520/p/13634002.html

posted on 2020-09-08 16:09  月满清爵  阅读(987)  评论(0编辑  收藏  举报