实现简单的LRU Cache

LRU Cache需要提供的功能:

  1. 有存储上限
  2. set: 保存键值对Entry,若空间已满,移除最久没有使用的一个Entry
  3. get: 根据键查找值,并将键调整为最近使用
  4. 对时间的要求:
    1. O(1)时间内保存和调整顺序
    2. O(1)时间内查找和调整顺序
public class LRUCache<K, V> {

    //使用双向链表
    private static class Node<K, V> {
        //key是为了删除节点时,根据key在Map中删除Entry
        K key; 
        V value;
        Node<K, V> pre;
        Node<K, V> next;

        public Node() {}

        public Node(K key, V value) {
            this.key = key;
            this.value = value;
        }
    }

    private int size;
    //使用Map保存所有缓存的内容,get的时间复杂度为O(1)
    //value为指向双向链表的指针,可以O(1)寻找节点在双向链表中的位置
    private Map<K, Node<K, V>> cache = new HashMap<>();

    //使用双向链表保存顺序,调整顺序的时间复杂度为O(1)
    private Node<K, V> head = new Node<>();
    private Node<K, V> tail = new Node<>();

    public LRUCache(int size) {
        if (size <= 0) {
            throw new IllegalArgumentException("Cache size must bigger than 0");
        }
        this.size = size;
        head.next = tail;
        tail.pre = head;
    }

    public void set(K k, V v) {
        Node<K, V> old = cache.get(k);
        //没有旧数据,构建Node,put进Cache,插入到链表的头部
        if (old == null) {
            //如果Cache空间已满,删掉最后一个节点(最久没有使用的节点)
            if (cache.size() == size) {
                Node<K, V> remove = removeLast();
                cache.remove(remove.key);
            }
            Node<K, V> node = new Node<>(k, v);
            cache.put(k, node);
            addFirst(node);
        } else {
            old.value = v;
            moveToFirst(old);
        }
    }

    public V get(K k) {
        Node<K, V> node = cache.get(k);
        if (node == null) {
            return null;
        }
        moveToFirst(node);
        return node.value;
    }

    private void moveToFirst(Node<K, V> node) {
        if (node.pre == head) {
            return;
        }
        removeNode(node);
        addFirst(node);
    }

    private void addFirst(Node<K, V> node) {
        if (node.pre == head) {
            return;
        }
        node.next = head.next;
        node.pre = head;
        head.next = node;
        node.next.pre = node;
    }

    private void removeNode(Node<K, V> node) {
        node.pre.next = node.next;
        node.next.pre = node.pre;
    }

    private Node<K, V> removeLast() {
        Node<K, V> remove = tail.pre;
        tail.pre = remove.pre;
        remove.pre.next = tail;
        return remove;
    }

    @Override
    public String toString() {
        List<K> ks = new ArrayList<>(cache.size());
        Node<K, V> n = head.next;
        while (n != tail) {
            ks.add(n.key);
            n = n.next;
        }
        return ks.toString();
    }

}

测试数据:

        LRUCache<Integer, Integer> cache = new LRUCache<>(3);
        System.out.println("======测试set======");
        cache.set(1, 1);
        System.out.println("after set 1: " + cache);
        cache.set(2, 2);
        System.out.println("after set 2: " + cache);
        cache.set(3, 3);
        System.out.println("after set 3: " + cache);
        cache.set(4, 4);
        System.out.println("after set 4: " + cache);
        cache.set(1, 1);
        System.out.println("after set 1: " + cache);

        System.out.println("\n======测试get======");

        System.out.println("get 1 = " + cache.get(1));
        System.out.println("after get 1: " + cache);
        System.out.println("get 3 = " + cache.get(3));
        System.out.println("after get 3: " + cache);
        System.out.println("get 4 = " + cache.get(4));
        System.out.println("after get 4: " + cache);

        System.out.println("\n======测试更改======");
        System.out.println("get 1: " + cache.get(1));
        cache.set(1, 11);
        System.out.println("after set 1: " + cache);
        System.out.println("get 1: " + cache.get(1));
        

测试结果:

======测试set======
after set 1: [1]
after set 2: [2, 1]
after set 3: [3, 2, 1]
after set 4: [4, 3, 2]
after set 1: [1, 4, 3]

======测试get======
get 1 = 1
after get 1: [1, 4, 3]
get 3 = 3
after get 3: [3, 1, 4]
get 4 = 4
after get 4: [4, 3, 1]

======测试更改======
get 1: 1
after set 1: [1, 4, 3]
get 1: 11

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,175评论 5 466
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,674评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,151评论 0 328
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,597评论 1 269
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,505评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 47,969评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,455评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,118评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,227评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,213评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,214评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,928评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,512评论 3 302
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,616评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,848评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,228评论 2 344
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,772评论 2 339

推荐阅读更多精彩内容