面试挂在了 LRU 缓存算法设计上

时间 2019-12-10

原文原文链接

好吧，有人可能以为我标题党了，但我想告诉大家的是，前阵子面试确实挂在了 RLU 缓存算法的设计上了。当时作题的时候，本身想的太多了，感受设计一个 LRU(Least recently used) 缓存算法，不会这么简单啊，因而理解错了题意（我也是服了，还能理解成这样，，，，），本身一波操做写了好多代码，后来卡住了，再去仔细看题，发现本身应该是理解错了，就是这么简单，设计一个 LRU 缓存算法。node

不过这时时间就很紧了，按道理若是你真的对这个算法很熟，十分钟就能写出来了，可是，本身虽然理解 LRU 缓存算法的思想，也知道具体步骤，但以前却历来没有去动手写过，致使在写的时候，很是不熟练，也就是说，你感受本身会和你可以用代码完美着写出来是彻底不是一回事，因此在此提醒各位，若是能够，必定要本身用代码实现一遍本身自觉得会的东西。千万不要以为本身理解了思想，就不用去写代码了，独自撸一遍代码，才是真的理解了。面试

今天我带你们用代码来实现一遍 LRU 缓存算法，之后你在遇到这类型的题，保证你完美秒杀它。算法

题目描述

设计并实现最不常用（LFU）缓存的数据结构。它应该支持如下操做：get 和 put。缓存

get(key) - 若是键存在于缓存中，则获取键的值（老是正数），不然返回 -1。数据结构

put(key, value) - 若是键不存在，请设置或插入值。当缓存达到其容量时，它应该在插入新项目以前，
使最不常用的项目无效。在此问题中，当存在平局（即两个或更多个键具备相同使用频率）时，
最近最少使用的键将被去除。this

进阶：.net

你是否能够在 O(1) 时间复杂度内执行两项操做？设计

示例：code

LFUCache cache = new LFUCache( 2 /* capacity (缓存容量) */ );



cache.put(1, 1);

cache.put(2, 2);

cache.get(1);       // 返回 1

cache.put(3, 3);    // 去除 key 2

cache.get(2);       // 返回 -1 (未找到key 2)

cache.get(3);       // 返回 3

cache.put(4, 4);    // 去除 key 1

cache.get(1);       // 返回 -1 (未找到 key 1)

cache.get(3);       // 返回 3

cache.get(4);       // 返回 4

基础版：单链表来解决

咱们要删的是最近最少使用的节点，一种比较容易想到的方法就是使用单链表这种数据结构来存储了。当咱们进行 put 操做的时候，会出现如下几种状况：ci

一、若是要 put(key,value) 已经存在于链表之中了（根据key来判断），那么咱们须要把链表中久的数据删除，而后把新的数据插入到链表的头部。、

二、若是要 put(key,value) 的数据没有存在于链表以后，咱们咱们须要判断下缓存区是否已满，若是满的话，则把链表尾部的节点删除，以后把新的数据插入到链表头部。若是没有满的话，直接把数据插入链表头部便可。

对于 get 操做，则会出现如下状况

一、若是要 get(key) 的数据存在于链表中，则把 value 返回，而且把该节点删除，删除以后把它插入到链表的头部。

二、若是要 get(key) 的数据不存在于链表以后，则直接返回 -1 便可。

大概的思路就是这样，不要以为很简单，让你手写的话，十分钟你不必定手写的出来。具体的代码，为了避免影响阅读，我在文章的最后面在放出来。

时间、空间复杂度分析

对于这种方法，put 和 get 都须要遍历链表查找数据是否存在，因此时间复杂度为 O(n)。空间复杂度为 O(1)。

空间换时间

在实际的应用中，当咱们要去读取一个数据的时候，会先判断该数据是否存在于缓存器中，若是存在，则返回，若是不存在，则去别的地方查找该数据（例如磁盘），找到后在把该数据存放于缓存器中，在返回。

因此在实际的应用中，put 操做通常伴随着 get 操做，也就是说，get 操做的次数是比较多的，并且命中率也是相对比较高的，进而 put 操做的次数是比较少的，咱们咱们是能够考虑采用空间换时间的方式来加快咱们的 get 的操做的。

例如咱们能够用一个额外哈希表（例如HashMap）来存放 key-value，这样的话，咱们的 get 操做就能够在 O(1) 的时间内寻找到目标节点，而且把 value 返回了。

然而，你们想一下，用了哈希表以后，get 操做真的可以在 O(1) 时间内完成吗？

用了哈希表以后，虽然咱们可以在 O(1) 时间内找到目标元素，能够，咱们还须要删除该元素，而且把该元素插入到链表头部啊，删除一个元素，咱们是须要定位到这个元素的前驱的，而后定位到这个元素的前驱，是须要 O(n) 时间复杂度的。

最后的结果是，用了哈希表时候，最坏时间复杂度仍是 O(1)，而空间复杂度也变为了 O(n)。

双向链表+哈希表

咱们都已经可以在 O(1) 时间复杂度找到要删除的节点了，之因此还得花 O(n) 时间复杂度才能删除，主要是时间是花在了节点前驱的查找上，为了解决这个问题，其实，咱们能够把单链表换成双链表，这样的话，咱们就能够很好着解决这个问题了，并且，换成双链表以后，你会发现，它要比单链表的操做简单多了。

因此咱们最后的方案是：双链表 + 哈希表，采用这两种数据结构的组合，咱们的 get 操做就能够在 O(1) 时间复杂度内完成了。因为 put 操做咱们要删除的节点通常是尾部节点，因此咱们能够用一个变量 tai 时刻记录尾部节点的位置，这样的话，咱们的 put 操做也能够在 O(1) 时间内完成了。

具体代码以下：

// 链表节点的定义
class LRUNode{
    String key;
    Object value;
    LRUNode next;
    LRUNode pre;

    public LRUNode(String key, Object value) {
        this.key = key;
        this.value = value;
    }
}

// LRU
public class LRUCache {
    Map<String, LRUNode> map = new HashMap<>();
    RLUNode head;
    RLUNode tail;
    // 缓存最大容量，咱们假设最大容量大于 1，
    // 固然，小于等于1的话须要多加一些判断另行处理
    int capacity;

    public RLUCache(int capacity) {
        this.capacity = capacity;
    }

    public void put(String key, Object value) {
        if (head == null) {
            head = new LRUNode(key, value);
            tail = head;
            map.put(key, head);
        }
        LRUNode node = map.get(key);
        if (node != null) {
            // 更新值
            node.value = value;
            // 把他从链表删除而且插入到头结点
            removeAndInsert(node);
        } else {
            LRUNode tmp = new LRUNode(key, value);
            // 若是会溢出
            if (map.size() >= capacity) {
                // 先把它从哈希表中删除
                map.remove(tail);
                // 删除尾部节点
                tail = tail.pre;
                tail.next = null;
            }
            map.put(key, tmp);
            // 插入
            tmp.next = head;
            head.pre = tmp;
            head = tmp;
        }
    }

    public Object get(String key) {
        LRUNode node = map.get(key);
        if (node != null) {
            // 把这个节点删除并插入到头结点
            removeAndInsert(node);
            return node.value;
        }
        return null;
    }
    private void removeAndInsert(LRUNode node) {
        // 特殊状况先判断，例如该节点是头结点或是尾部节点
        if (node == head) {
            return;
        } else if (node == tail) {
            tail = node.pre;
            tail.next = null;
        } else {
            node.pre.next = node.next;
            node.next.pre = node.pre;
        }
        // 插入到头结点
        node.next = head;
        node.pre = null;
        head.pre = node;
        head = node;
    }
}

这里须要提醒的是，对于链表这种数据结构，头结点和尾节点是两个比较特殊的点，若是要删除的节点是头结点或者尾节点，咱们通常要先对他们进行处理。

这里放一下单链表版本的吧

// 定义链表节点
class RLUNode{
    String key;
    Object value;
    RLUNode next;

    public RLUNode(String key, Object value) {
        this.key = key;
        this.value = value;
    }
}
// 把名字写错了，把 LRU写成了RLU
public class RLUCache {
    RLUNode head;
    int size = 0;// 当前大小
    int capacity = 0; // 最大容量

    public RLUCache(int capacity) {
        this.capacity = capacity;
    }

    public Object get(String key) {
        RLUNode cur = head;
        RLUNode pre = head;// 指向要删除节点的前驱
        // 找到对应的节点，并把对应的节点放在链表头部
        // 先考虑特殊状况
        if(head == null)
            return null;
        if(cur.key.equals(key))
            return cur.value;
        // 进行查找
        cur = cur.next;
        while (cur != null) {
            if (cur.key.equals(key)) {
                break;
            }
            pre = cur;
            cur = cur.next;
        }
        // 表明没找到了节点
        if (cur == null)
            return null;

        // 进行删除
        pre.next = cur.next;
        // 删除以后插入头结点
        cur.next = head;
        head = cur;
        return cur.value;
    }

    public void put(String key, Object value) {
        // 若是最大容量是 1，那就没办法了，，，，，
        if (capacity == 1) {
            head = new RLUNode(key, value);
        }
        RLUNode cur = head;
        RLUNode pre = head;
        // 先查看链表是否为空
        if (head == null) {
            head = new RLUNode(key, value);
            return;
        }
        // 先查看该节点是否存在
        // 第一个节点比较特殊，先进行判断
        if (head.key.equals(key)) {
            head.value = value;
            return;
        }
        cur = cur.next;
        while (cur != null) {
            if (cur.key.equals(key)) {
                break;
            }
            pre = cur;
            cur = cur.next;
        }
        // 表明要插入的节点的 key 已存在，则进行 value 的更新
        // 以及把它放到第一个节点去
        if (cur != null) {
            cur.value = value;
            pre.next = cur.next;
            cur.next = head;
            head = cur;
        } else {
            // 先建立一个节点
            RLUNode tmp = new RLUNode(key, value);
            // 该节点不存在，须要判断插入后会不会溢出
            if (size >= capacity) {
                // 直接把最后一个节点移除
                cur = head;
                while (cur.next != null && cur.next.next != null) {
                    cur = cur.next;
                }
                cur.next = null;
                tmp.next = head;
                head = tmp;
            }
        }
    }
}

若是要时间，强烈建议本身手动实现一波。

最后推广下个人公众号：苦逼的码农：戳我便可关注，文章都会首发于个人公众号，期待各路英雄的关注交流。