LeetCode 460 LFU缓存 NC94 设计LFU缓存结构

BM101 设计LFU缓存结构

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

实现 LFUCache 类：

LFUCache(int capacity) - 用数据结构的容量 capacity 初始化对象
int get(int key) - 如果键存在于缓存中，则获取键的值，否则返回 -1。
void put(int key, int value) - 如果键已存在，则变更其值；如果键不存在，请插入键值对。当缓存达到其容量时，则应该在插入新项之前，使最不经常使用的项无效。在此问题中，当存在平局（即两个或更多个键具有相同使用频率）时，应该去除 最久未使用 的键。
注意「项的使用次数」就是自插入该项以来对其调用 get 和 put 函数的次数之和。使用次数会在对应项被移除后置为 0 。

进阶：

你是否可以在 O(1) 时间复杂度内执行两项操作？
 

示例：

输入：
["LFUCache", "put", "put", "get", "put", "get", "get", "put", "get", "get", "get"]
[[2], [1, 1], [2, 2], [1], [3, 3], [2], [3], [4, 4], [1], [3], [4]]
输出：
[null, null, null, 1, null, -1, 3, null, -1, 3, 4]

解释：
LFUCache lFUCache = new LFUCache(2);
lFUCache.put(1, 1);
lFUCache.put(2, 2);
lFUCache.get(1);      // 返回 1
lFUCache.put(3, 3);   // 去除键 2
lFUCache.get(2);      // 返回 -1（未找到）
lFUCache.get(3);      // 返回 3
lFUCache.put(4, 4);   // 去除键 1
lFUCache.get(1);      // 返回 -1（未找到）
lFUCache.get(3);      // 返回 3
lFUCache.get(4);      // 返回 4
 

提示：

0 <= capacity, key, value <= 104
最多调用 105 次 get 和 put 方法

解题思路：

双哈希表

力扣

key与freq映射，为了找key在哪个频率下；
key与value映射，为了找key对应的value值，这个需要一个有序字典OrderedDict，为了好弹出同一个频率下，最开始加入的key（当然是容量不够，需要弹出的时候）。
对于第二个字典，可以用python内建有序字典OrderedDict实现。

Python代码：

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# lfu design
# @param operators int整型二维数组 ops
# @param k int整型 the k
# @return int整型一维数组
#
from collections import OrderedDict, defaultdict

class LFUCache:
    def __init__(self, k):
        self.k2f = dict()
        self.f2kv = defaultdict(OrderedDict)
        self.k = k
        self.min_f = 0

    def get(self, key):
        if key not in self.k2f:
            return -1
        else:
            f = self.k2f[key]
            value = self.f2kv[f].pop(key)
            if self.min_f == f and not self.f2kv[f]:
                self.min_f += 1
            self.k2f[key] = f + 1
            self.f2kv[f+1][key] = value
            return value
        
    def set(self, key, value):
        if key not in self.k2f:
            if self.k == len(self.k2f):
                k, v = self.f2kv[self.min_f].popitem(last=False)
                self.k2f.pop(k)
            self.k2f[key] = 1
            self.f2kv[1][key] = value
            self.min_f = 1
        else:
            f = self.k2f[key]
            self.f2kv[f].pop(key)
            if not self.f2kv[f] and self.min_f == f:
                self.min_f += 1
            self.k2f[key] = f + 1
            self.f2kv[f + 1][key] = value

            
class Solution:
    def LFU(self , operators: List[List[int]], k: int) -> List[int]:
        # write code here
        res = []
        lfu_cache = LFUCache(k)
        for operator in operators:
            if operator[0] == 1:
                lfu_cache.set(operator[1], operator[2])
            else:
                res.append(lfu_cache.get(operator[1]))
        return res