LFU缓存设计

题目描述

请你为 最不经常使用（LFU）缓存算法设计并实现数据结构。

实现 LFUCache 类：

LFUCache(int capacity) - 用数据结构的容量 capacity 初始化对象
int get(int key) - 如果键 key 存在于缓存中，则获取键的值，否则返回 -1 。
void put(int key, int value) - 如果键 key 已存在，则变更其值；如果键不存在，请插入键值对。当缓存达到其容量 capacity 时，则应该在插入新项之前，移除最不经常使用的项。在此问题中，当存在平局（即两个或更多个键具有相同使用频率）时，应该去除 最久未使用 的键。
为了确定最不常使用的键，可以为缓存中的每个键维护一个 使用计数器 。使用计数最小的键是最久未使用的键。

当一个键首次插入到缓存中时，它的使用计数器被设置为 1 (由于 put 操作)。对缓存中的键执行 get 或 put 操作，使用计数器的值将会递增。

函数 get 和 put 必须以 O(1) 的平均时间复杂度运行。

设计思路

我们定义两个哈希表，第一个 freq_table 以频率 freq 为索引，每个索引存放一个双向链表，这个链表里存放所有使用频率为 freq 的缓存，缓存里存放三个信息，分别为键 key，值 value，以及使用频率 freq。第二个 data 以键值 key 为索引，每个索引存放对应缓存在 freq_table 中链表里的内存地址，这样我们就能利用两个哈希表来使得两个操作的时间复杂度均为 O(1)。同时需要记录一个当前缓存最少使用的频率 minFreq，这是为了删除操作服务的。

对于 get(key) 操作，我们能通过索引 key 在 data 中找到缓存在 freq_table 中的链表的内存地址，如果不存在直接返回 -1，否则我们能获取到对应缓存的相关信息，这样我们就能知道缓存的键值还有使用频率，直接返回 key 对应的值即可。

但是我们注意到 get 操作后这个缓存的使用频率加一了，所以我们需要更新缓存在哈希表 freq_table 中的位置。已知这个缓存的键 key，值 value，以及使用频率 freq，那么该缓存应该存放到 freq_table 中 freq + 1 索引下的链表中。所以我们在当前链表中 O(1) 删除该缓存对应的节点，根据情况更新 minFreq 值，然后将其O(1) 插入到 freq + 1 索引下的链表头完成更新。这其中的操作复杂度均为 O(1)。你可能会疑惑更新的时候为什么是插入到链表头，这其实是为了保证缓存在当前链表中从链表头到链表尾的插入时间是有序的，为下面的删除操作服务。

对于 put(key, value) 操作，我们先通过索引 key在 data 中查看是否有对应的缓存，如果有的话，其实操作等价于 get(key) 操作，唯一的区别就是我们需要将当前的缓存里的值更新为 value。如果没有的话，相当于是新加入的缓存，如果缓存已经到达容量，需要先删除最近最少使用的缓存，再进行插入。

先考虑插入，由于是新插入的，所以缓存的使用频率一定是 1，所以我们将缓存的信息插入到 freq_table 中 1 索引下的列表头即可，同时更新 key_table[key] 的信息，以及更新 minFreq = 1。

那么剩下的就是删除操作了，由于我们实时维护了 minFreq，所以我们能够知道 freq_table 里目前最少使用频率的索引，同时因为我们保证了链表中从链表头到链表尾的插入时间是有序的，所以 freq_table[minFreq] 的链表中链表尾的节点即为使用频率最小且插入时间最早的节点，我们删除它同时根据情况更新 minFreq ，整个时间复杂度均为 O(1)。

代码实现：

  Map<Integer, Node> data;

    Map<Integer, LinkedHashSet<Node>> fre;

    int size;

    int min;


    public LFUCache(int capacity) {
        this.size = capacity;
        data = new HashMap<>();
        fre = new TreeMap<>();

    }

    public int get(int key) {
        Node v = data.get(key);
        if (v == null) {
            return -1;
        } else {
            incFre(v);
            return v.val;
        }
    }

    public void put(
            int key, int value) {
        Node v = data.get(key);
        if (v == null) {
            if (size == data.size()) {
                removeLeast();
            }
            data.put(key, new Node(key, value, 1));
            min = 1;
            LinkedHashSet<Node> freq = fre.getOrDefault(1, new LinkedHashSet<>());
            freq.add(data.get(key));
            fre.put(1, freq);
        } else {
            v.setVal(value);
            incFre(v);
        }

    }

    private void incFre(Node v) {
        LinkedHashSet<Node> freq = fre.get(v.times);
        freq.remove(v);
        if (freq.isEmpty()) {
            fre.remove(v.times);
            if (min == v.times) {
                min = min + 1;
            }
        }
        LinkedHashSet<Node> moreFreq = fre.getOrDefault(v.times + 1, new LinkedHashSet<>());
        moreFreq.add(v);
        fre.put(v.times + 1, moreFreq);
        v.setTimes(v.times + 1);
    }

    private void removeLeast() {
        LinkedHashSet<Node> freq = fre.get(min);
        Node e = freq.iterator().next();
        freq.remove(e);
        data.remove(e.key);
        if (freq.isEmpty()) {
            fre.remove(min);
            min = fre.isEmpty() ? 1 : fre.keySet().iterator().next();
        }

    }

    class Node {
        int key;

        int val;
        int times;


        public Node(int key, int val, int t) {
            this.key = key;
            this.val = val;
            this.times = t;
        }

        public void setTimes(int times) {
            this.times = times;
        }

        public void setVal(int val) {
            this.val = val;
        }
    }