牛客-LRU缓存结构

题目

描述

设计LRU(最近最少使用)缓存结构，该结构在构造时确定大小，假设大小为 k ，并有如下两个功能

1. set(key, value)：将记录(key, value)插入该结构
2. get(key)：返回key对应的value值

提示:
1.某个key的set或get操作一旦发生，认为这个key的记录成了最常使用的，然后都会刷新缓存。
2.当缓存的大小超过k时，移除最不经常使用的记录。
3.输入一个二维数组与k，二维数组每一维有2个或者3个数字，第1个数字为opt，第2，3个数字为key，value
若opt=1，接下来两个整数key, value，表示set(key, value)
若opt=2，接下来一个整数key，表示get(key)，若key未出现过或已被移除，则返回-1
对于每个opt=2，输出一个答案
4.为了方便区分缓存里key与value，下面说明的缓存里key用""号包裹

要求：set和get操作复杂度均为 O(1)

输入：
[[1,1,1],[1,2,2],[1,3,2],[2,1],[1,4,4],[2,2]],3
返回值：
[1,-1]
说明：
[1,1,1]，第一个1表示opt=1，要set(1,1)，即将(1,1)插入缓存，缓存是{“1”=1}
[1,2,2]，第一个1表示opt=1，要set(2,2)，即将(2,2)插入缓存，缓存是{“1”=1,“2”=2}
[1,3,2]，第一个1表示opt=1，要set(3,2)，即将(3,2)插入缓存，缓存是{“1”=1,“2”=2,“3”=2}
[2,1]，第一个2表示opt=2，要get(1)，返回是[1]，因为get(1)操作，缓存更新，缓存是{“2”=2,“3”=2,“1”=1}
[1,4,4]，第一个1表示opt=1，要set(4,4)，即将(4,4)插入缓存，但是缓存已经达到最大容量3，移除最不经常使用的{“2”=2}，插入{“4”=4}，缓存是{“3”=2,“1”=1,“4”=4}
[2,2]，第一个2表示opt=2，要get(2)，查找不到，返回是[1,-1]

思路

主要是理解题目，get之后要更新状态，get的key和value变成了最近使用，如果没有get到key，则返回-1。set的时候要考虑有没有超过最大的缓存限制，超过的删除时间线上最不常使用的key和value。用list来操作这种更新很方便。

代码

python版本：

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# lru design
# @param operators int整型二维数组 the ops
# @param k int整型 the k
# @return int整型一维数组
#
class Solution:
    lru_list = []
    def LRU(self , operators: List[List[int]], k: int) -> List[int]:
        # write code here
        res = []
        for op in operators:
            if(len(op)==3):
                assert op[0]==1
                kv = op[1:]
                kv[0] = str(kv[0])
                self.set(kv, k)
                
            elif(len(op)==2):
                assert op[0]==2
                res.append(self.get(str(op[1])))
        return res
    
    def get(self, key):
        lru_dict = dict(self.lru_list)
        res = -1
        if key in lru_dict.keys():
            res = lru_dict[key]
            self.lru_list.remove([key, res])
            self.lru_list.append([key, res])
        return res
    
    def set(self, kv, k):
        if(len(self.lru_list))>=k:
            self.lru_list.pop(0)
        self.lru_list.append(kv)

c++版本：

无