-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path146.py
48 lines (41 loc) · 1.4 KB
/
146.py
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
36
37
38
39
40
41
42
43
44
45
46
47
48
'''
@Date: 2020-07-31 15:28:25
@LastEditors: FrankCast1e
@LastEditTime: 2020-07-31 15:55:47
@FilePath: /LeetCode/146.py
'''
class LRUCache:
def __init__(self, capacity: int):
self.upper_bound = capacity
self.storage_dict = dict()
self.use_situa_queue = []
def get(self, key: int) -> int:
if key in self.storage_dict:
if key in self.use_situa_queue:
index = self.use_situa_queue.index(key)
del self.use_situa_queue[index]
self.use_situa_queue.append(key)
return self.storage_dict[key]
else:
return -1
def put(self, key: int, value: int) -> None:
if key not in self.storage_dict and len(self.storage_dict) >= self.upper_bound:
least_recent_used = self.use_situa_queue[0]
del self.use_situa_queue[0]
del self.storage_dict[least_recent_used]
self.storage_dict[key] = value
if key in self.use_situa_queue:
index = self.use_situa_queue.index(key)
del self.use_situa_queue[index]
self.use_situa_queue.append(key)
# Your LRUCache object will be instantiated and called as such:
obj = LRUCache(2)
obj.put(1, 1)
obj.put(2, 2)
print(obj.get(1))
obj.put(3, 3)
print(obj.get(2))
obj.put(4, 4)
print(obj.get(1))
print(obj.get(3))
print(obj.get(4))