牛客网-NC93 设计LRU缓存结构


描述

设计LRU(最近最少使用)缓存结构,该结构在构造时确定大小,假设大小为 capacity ,操作次数是 n ,并有如下功能:

  1. Solution(int capacity) 以正整数作为容量 capacity 初始化 LRU 缓存

  2. get(key):如果关键字 key 存在于缓存中,则返回key对应的value值,否则返回 -1 。

  3. set(key, value):将记录(key, value)插入该结构,如果关键字 key 已经存在,则变更其数据值 value,如果不存在,则向缓存中插入该组 key-value ,如果key-value的数量超过capacity,弹出最久未使用的key-value

提示:

1.某个key的set或get操作一旦发生则认为这个key的记录成了最常使用的然后都会刷新缓存
2.当缓存的大小超过capacity时移除最不经常使用的记录
3.返回的value都以字符串形式表达如果是set则会输出"null"来表示(不需要用户返回系统会自动输出)方便观察
4.函数set和get必须以O(1)的方式运行
5.为了方便区分缓存里key与value下面说明的缓存里key用""号包裹

数据范围:

1<= capacity<=10^5
0<= key,val <= 10^9
1<= n<= 10^5

方法一

class Solution:

    def __init__(self, capacity: int):
    # write code here
        self.capacity = capacity
        self.lru_dict = dict()
        self.lru_list = list()

    def get(self, key: int) -> int:
        # write code here
        if key not in self.lru_list:
            return -1
        self.lru_list.remove(key)
        self.lru_list.append(key)
        return self.lru_dict[key]

    def set(self, key: int, value: int) -> None:
        # write code here
        if len(self.lru_list) < self.capacity and key not in self.lru_list:
            self.lru_list.append(key)
        else:
            pop_key = self.lru_list.pop(0)
            self.lru_dict.pop(pop_key)
            self.lru_list.append(key)
        self.lru_dict[key] = value
        return None