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)
大家可以根据思路,自己敲敲代码,有不明白的留言问我哦~