leetcode 146. LRUCache (用Java实现)

LRUCache 全称为 Least Recently Used Cache ,用Java实现的话,可以很简单地用LinkedHashMap来实现。但如果面试过程中碰到面试官出这道题,大概率是希望你不借助过多的工具类,而是把一个LRUCache内部的get、put等方法给写一遍,以证实你是真的理解了。。

首先明确我们需要用到的数据结构:

1、hash表

为保证时间复杂度在O(1),hash表是必不可少的,它的作用就是充当缓存的容器;

2、双向链表

双向链表的每个节点都具备指向前驱节点和指向后驱节点的两个指针,其双向查询的特点使得我们能够让一个(新/旧)节点以O(1)的时间复杂度快速入到链表的头部或尾部。而如果只是单向的链表,我们无法保证O(1)的时间复杂度的(比如删除队尾节点的操作,即使一开始定义了队尾节点,但是因为其单向的特点,删除时还是需要从头节点遍历到队尾的前一个节点,让其next指针指向空,才能将队尾删除)。

接下来一起看下代码中的逻辑:

  • 构造器

1、我们将构造器传进来的capacity参数作为缓存允许的最大容量,同时初始化当前缓存的数量size为0;

2、定义头部节点head和尾部节点tail两个dummyNode,这是链表类题目的常用技巧,为的是避免处理头节点为空的边界问题,同时要初始化好head和tail节点互相指向的关联关系;

  • get方法

根据key查询是否存在于缓存之中,不存在则直接返回null;如果存在,我们就要维护一下链表,将该节点移动到链表头部(同理,用尾部来作为最新节点的存放位置也是可以的),表示该节点最近被访问到了。

  • put方法

判断该key是否存在于缓存之中

1、如果不存在

  • 首先先新建一个节点放入hash表中,同时将这个节点添加到链表的头部,表示该节点最近被访问到了;
  • 放入缓存后,我们要判断加上这个节点后,缓存的容量是否超过设定的最大容量,如果是,就要删掉链表中的尾部节点,并根据尾部节点的key,将其从hash表中remove掉

2、如果存在

  • 移动该节点到链表头部,并更新节点中的value即可。

除此之外,对于链表节点的添加、移动、删除等操作,我们最好将方法单独抽离开来,这样做好处有两个:

一是可以对抽离开的方法进行复用;

二是在写get、put方法时,可以专注于缓存整体的维护逻辑,而不用关注到链表这一层数据结构,降低懵逼的风险。。

代码如下

import java.util.HashMap;
import java.util.Map;

public class LRUCache {

    Map<Integer, DLinkedNode> cache ;
    int size;
    int capacity;
    DLinkedNode head, tail ;

    public static void main(String[] args) {
        LRUCache lruCache = new LRUCache(3);
        lruCache.put(1, 2);
        lruCache.put(2, 2);
        lruCache.put(3, 2);
        lruCache.get(1);
        lruCache.put(4, 2);
        System.out.println(lruCache.cache.keySet());// [1, 3, 4]
    }

    public LRUCache(int capacity) {
        this.size = 0;
        this.capacity = capacity;
        cache = new HashMap<>();
        // 维护两个dummy node(始终在头尾)
        this.head = new DLinkedNode();
        this.tail = new DLinkedNode();
        head.next = tail;
        tail.prev = head;
    }

    public int get(int key) {
        DLinkedNode node = cache.get(key);
        if (node == null) {
            return -1;
        }
        moveToHead(node);
        return node.v;
    }

    public void put(int key, int value) {
        DLinkedNode node = cache.get(key);
        if (node == null) {
            node = new DLinkedNode(key, value);
            cache.put(key, node);
            // 移动到头部
            addToHead(node);
            if (++size > capacity) {
                // 丢弃tail
                DLinkedNode tail = removeTail();
                cache.remove(tail.k);
                --size;
            }
        } else {
            // 替换并放到头部
            node.v = value;
            moveToHead(node);
        }
    }

    /**
     * 将节点移动到头部
     */
    private void moveToHead(DLinkedNode node) {
        removeNode(node);
        addToHead(node);
    }

    /**
     * 删除节点
     */
    private void removeNode(DLinkedNode node) {
        node.next.prev = node.prev;
        node.prev.next = node.next;
    }

    /**
     * 添加到头部
     */
    private void addToHead(DLinkedNode node) {
        node.prev = head;
        node.next = head.next;
        head.next.prev = node;
        head.next = node;
    }

    /**
     * 删除尾部节点
     */
    private DLinkedNode removeTail() {
        DLinkedNode node = tail.prev;
        removeNode(node);
        return node;
    }

    /**
     * 双向链表
     */
    class DLinkedNode {
        DLinkedNode prev;
        DLinkedNode next;
        // key作用体现在干掉tail的时候可以通过node的key去remove哈希表中的元素
        int k;
        int v;

        public DLinkedNode() {
        }

        public DLinkedNode(int k, int v) {
            this.k = k;
            this.v = v;
        }

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

推荐阅读更多精彩内容