【题目】
【代码】
执行用时:6888 ms, 在所有 Python3 提交中击败了5.00% 的用户
内存消耗:71.1 MB, 在所有 Python3 提交中击败了90.79% 的用户
通过测试用例:22 / 22
class LRUCache:
def __init__(self, capacity: int):
self.key=[]
self.value=[]
self.visit=[]
self.cnt=0
self.capacity=capacity
def get(self, key: int) -> int:
if key in self.key:
self.cnt+=1
self.visit[self.key.index(key)]=self.cnt
return self.value[self.key.index(key)]
return -1
def put(self, key: int, value: int) -> None:
if key in self.key:
index=self.key.index(key)
self.cnt+=1
self.visit[index]=self.cnt
self.value[index]=value
else:
if len(self.key)>=self.capacity:
min_visit=min(self.visit)
min_visit_pos=self.visit.index(min_visit)
self.key.pop(min_visit_pos)
self.value.pop(min_visit_pos)
self.visit.pop(min_visit_pos)
self.key.append(key)
self.cnt+=1
self.value.append(value)
self.visit.append(self.cnt)
【方法2】不讲武德 OrderedDict
执行用时:576 ms, 在所有 Python3 提交中击败了79.73% 的用户
内存消耗:71.7 MB, 在所有 Python3 提交中击败了50.14% 的用户
通过测试用例:22 / 22
class LRUCache(collections.OrderedDict):
def __init__(self, capcy: int):
super().__init__()
self.capacity = capcy
def get(self, ky: int) -> int:
if ky not in self:
return -1
self.move_to_end(ky)
return self[ky]
def put(self, key: int, value: int) -> None:
if key in self:
self.move_to_end(key)
self[key] = value
if len(self) > self.capacity:
self.popitem(last=False)