LRUCache简单实现

LRUCache简单实现


在这里插入图片描述

来源于LeetCode. 146. LRU 缓存
在这里插入图片描述
题目要求实现一个LRUCache,我们使用get时,需要额外把该节点移动到开头;

我们使用put时,如果存在该节点,我们需要把该节点删除,并且插入到开头;不存在则插入到开头,此时需要判断长度是否大于给定的capacity。

并且实现这两种操作时,时间复杂度都必须为O(1);get我们使用HashMap便可以达到我们的时间复杂度,但是HashMap不是有序的,但它的子类LinkedHashMap是有序的,所以我们使用LinkedHashMap便可以达到本题的效果,但是我们还是自己实现一个双向链表,才能考验代码功底

class LRUCache {
    
    
    private DoubleLinkedList head;
    private DoubleLinkedList tail;
    private int capacity;
    private Map<Integer,DoubleLinkedList> cache;

    public LRUCache(int capacity) {
    
    
        cache=new HashMap<>();
        this.capacity=capacity;
        head=new DoubleLinkedList(0,0);
        tail=new DoubleLinkedList(0,0);
        head.next=tail;
        tail.pre=head;
    }

    public int get(int key) {
    
    
        if(cache.containsKey(key)){
    
    
            DoubleLinkedList temp=cache.get(key);
            //删除掉这个节点
            delete(temp);
            //在头结点插入这个节点
            insert(temp);
            return temp.val;
        }else{
    
    
            return -1;
        }
    }

    public void put(int key, int value) {
    
    
        if(cache.containsKey(key)){
    
    
            DoubleLinkedList temp=cache.get(key);
            delete(temp);
            insert(temp);
            temp.val=value;
        }else{
    
    
            DoubleLinkedList temp=new DoubleLinkedList(key,value);
            insert(temp);
            if(cache.size()>capacity){
    
    
                delete(tail.pre);
            }
        }
    }

    public void delete(DoubleLinkedList temp){
    
    
        cache.remove(temp.key);
        temp.next.pre=temp.pre;
        temp.pre.next=temp.next;
    }

    public void insert(DoubleLinkedList temp){
    
    
        cache.put(temp.key,temp);
        temp.next=head.next;
        temp.pre=head;
        head.next.pre=temp;
        head.next=temp;
    }
    class DoubleLinkedList{
    
    
        DoubleLinkedList pre;
        DoubleLinkedList next;
        int key;
        int val;
        public DoubleLinkedList(int key,int val){
    
    
            this.key=key;
            this.val=val;
        }
    }
}

/**
 * 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);
 */

猜你喜欢

转载自blog.csdn.net/small_engineer/article/details/125111943
今日推荐