LRU缓存(java实现)

import java.util.HashMap;
import java.util.Iterator;

public class LRU<K,V> implements Iterable  {
    private Node head;
    private Node tail;
    private HashMap<K,Node> map;
    private int maxSize;
    private class Node<K,V>{
         Node pre;
         Node next;
         K k;
         V v;
         Node(K k,V v){
            this.k=k;
            this.v=v;
        }
    }
    private void unlink(Node node){
        Node preNode=node.pre;
        Node nextNode=node.next;
        preNode.next=nextNode;
        nextNode.pre=preNode;
        node.next=null;
        node.pre=null;
    }
    public V get(K k){
        if(!map.containsKey(k)){
            return null;
        }
        Node node = map.get(k);
        unlink(node);
        appendHead(node);
        return (V)node.v;
    }
    public void set(K k,V v){
        if(map.containsKey(k)){
            unlink(map.get(k));
        }
        Node<K, V> node = new Node<>(k, v);
        map.put(k,node);
        appendHead(node);
        if(map.size()>maxSize){
            removeOldNode();
        }
    }
    private void removeOldNode() {
        Node node=tail.pre;
        unlink(node);
        map.remove(node.k);
    }

    private void appendHead(Node node){
        Node oldNode=head.next;
        oldNode.pre=node;
        node.next=oldNode;
        head.next=node;
        node.pre=head;
    }
    public LRU(int maxSize){
        this.maxSize=maxSize;
        map=new HashMap<>(maxSize*4/3);
        head=new Node(null,null);
        tail=new Node(null,null);
        head.next=tail;
        tail.pre=head;
    }
    public LRU(){
        this.maxSize=20;
        map=new HashMap<>(maxSize*4/3);
        head=new Node(null,null);
        tail=new Node(null,null);
        head.next=tail;
        tail.pre=head;
    }
    @Override
    public Iterator<K> iterator() {
        return new Iterator<K>() {
            Node node=head.next;
            @Override
            public boolean hasNext() {
                return node!=tail;
            }
            @Override
            public K next() {
                K k = (K)node.k;
                node=node.next;
                return k;
            }
        };
    }
}
//继承LinkedHashMap实现更为简单
class LRULinkedHashMap<K,V> extends LinkedHashMap{
    private static final int MAX_ENTRIES=3;
    LRULinkedHashMap(){
        super(MAX_ENTRIES,0.75f,true);
    }
    @Override
    protected boolean removeEldestEntry(Map.Entry eldest){
        return size()>MAX_ENTRIES;
    }
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值