-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathLFUCache.py
More file actions
35 lines (33 loc) · 1.11 KB
/
LFUCache.py
File metadata and controls
35 lines (33 loc) · 1.11 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
class LFUCache:
def __init__(self, capacity: int):
self.capacity=capacity
self.minfreq=None
self.keyfreq={}
self.freqkeys=collections.defaultdict(collections.OrderedDict)
def get(self, key: int) -> int:
if key not in self.keyfreq:
return -1
freq=self.keyfreq[key]
val=self.freqkeys[freq][key]
del self.freqkeys[freq][key]
if not self.freqkeys[freq]:
if freq==self.minfreq:
self.minfreq+=1
del self.freqkeys[freq]
self.keyfreq[key]=freq+1
self.freqkeys[freq+1][key]=val
return val
def put(self, key: int, value: int) -> None:
if self.capacity<=0:
return
if key in self.keyfreq:
freq=self.keyfreq[key]
self.freqkeys[freq][key]=value
self.get(key)
return
if self.capacity==len(self.keyfreq):
delkey,delval=self.freqkeys[self.minfreq].popitem(last=False)
del self.keyfreq[delkey]
self.keyfreq[key]=1
self.freqkeys[1][key]=value
self.minfreq=1