python每日算法 | LFU缓存（Day5）

Day5 难度：困难
题干：
设计并实现最不经常使用（LFU）缓存的数据结构。它应该支持以下操作：get 和 put。
get(key) - 如果键存在于缓存中，则获取键的值（总是正数），否则返回 -1。
put(key, value) - 如果键不存在，请设置或插入值。当缓存达到其容量时，它应该在插入新项目之前，使最不经常使用的项目无效。在此问题中，当存在平局（即两个或更多个键具有相同使用频率）时，最近最少使用的键将被去除。

示例：

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

思路：

根据题目要求，缓存用字典表示，内容有[value, freq, time]值，频率，时间
在获取和设置之前，都应该先使time+1
在相同频率的时候，验证time
具体步骤，参考代码中注释

代码：

class LFUCache:

    def __init__(self, capacity: int):
        self.capacity = capacity        # 缓存容量
        self.time = 0                   # 时间计数，每次调用get()则+1
        self.obj = {}                   # key:[value, freq, time]
        
    def get(self, key: int) -> int:
        if key in self.obj:
            self.time += 1          # 使用时间+1
            self.obj[key][1] += 1   # 频率加1
            self.obj[key][2] = self.time
            return self.obj[key][0]
        else:
            return -1

    def put(self, key: int, value: int) -> None:
        if self.capacity == 0:      # 缓存容量为0
            return None
        self.time += 1
        if key in self.obj:         # key在obj中
            self.obj[key][0] = value
            self.obj[key][1] += 1
            self.obj[key][2] = self.time
            return None
        else:                       # key不在obj中
            if len(self.obj) < self.capacity:       # 缓存容量未满，直接添加key,value
                self.obj[key] = [value, 1, self.time]
            else:                                   # 缓存容量已满，找到频率最低的key
                less_freq = min(self.obj.values(), key=lambda x: x[1])[1]
                keys = [i for i ,j in self.obj.items() if j[1] == less_freq]
                if len(keys) == 1:                  # 只有一个频率最低的key，直接删除
                    del self.obj[keys[0]]
                else:                               # 有多个频率最低的，找出时间最小的，删除
                    less_key = keys[0]
                    less_time = self.obj[less_key][2]
                    for i in keys:
                        if self.obj[i][2] < less_time:
                            less_key = i
                            less_time = self.obj[less_key][2]
                    del self.obj[less_key]
                self.obj[key] = [value, 1, self.time]


# Your LFUCache object will be instantiated and called as such:
# obj = LFUCache(capacity)
# param_1 = obj.get(key)
# obj.put(key,value)

大家可以根据思路，自己敲敲代码，有不明白的留言问我哦~