[牛客题霸-研发] NC93-设计LRU缓存结构

----------------------------------------------本题链接----------------------------------------------

题目描述

设计LRU缓存结构，该结构在构造时确定大小，假设大小为K，并有如下两个功能：
set(key, value)：将记录(key, value)插入该结构
get(key)：返回key对应的value值

[要求]
set和get方法的时间复杂度为O(1)
某个key的set或get操作一旦发生，认为这个key的记录成了最常使用的。
当缓存的大小超过K时，移除最不经常使用的记录，即set或get最久远的。
若opt=1，接下来两个整数x, y，表示set(x, y)
若opt=2，接下来一个整数x，表示get(x)，若x未出现过或已被移除，则返回-1
对于每个操作2，输出一个答案

示例

输入
[[1,1,1],[1,2,2],[1,3,2],[2,1],[1,4,4],[2,2]],3
 
返回值
[1,-1]

说明：
第一次操作后：最常使用的记录为(“1”, 1)
第二次操作后：最常使用的记录为(“2”, 2)，(“1”, 1)变为最不常用的
第三次操作后：最常使用的记录为(“3”, 2)，(“1”, 1)还是最不常用的
第四次操作后：最常用的记录为(“1”, 1)，(“2”, 2)变为最不常用的
第五次操作后：大小超过了3，所以移除此时最不常使用的记录(“2”, 2)，加入记录(“4”, 4)，并且为最常使用的记录，然后(“3”, 2)变为最不常使用的记录

思路

LRU = Least Recently Used （最近最少使用）
很容易想到Map操作，关键点在于：

1. map容量达到k值的时候，set需要删除最初不常用的值，同时put进去最新的值
2. get获取map里的值，如果存在则返回key对应value，且该值变为最常用

这些要求让人联想到队列操作：先进入map的被挤出队列，get获取到就再让该值入队列一次（这样可以让头部键值对总为最不常用）
从而考虑用顺序存放的LinkedHashMap来操作（就是时间和空间复杂度有点高）

算法过程

1. 判断异常情况
2. 遍历数组判断操作
   2.1 set: 没有超过k值，则存入；超过k值，则删除Map开头，再存入
   2.2 get: map有key，则保存对应value值，并将该键值对先删除再存入；map没有key，返回-1
3. 输出结果

解答

public class Solution {
    /**
     * lru design
     * @param operators int整型二维数组 the ops
     * @param k int整型 the k
     * @return int整型一维数组
     */
    private Map<Integer, Integer> lruMap = new LinkedHashMap<>();
    private ArrayList<Integer> res = new ArrayList();
    private int k;
    
    public int[] LRU (int[][] operators, int k) {
        if(operators.length == 0 || k == 0)
            return null;
        this.k = k;
        
        for(int i = 0; i < operators.length; i++){
            if(operators[i][0] == 1){
                set(operators[i][1], operators[i][2]);
            } else {
                get(operators[i][1]);
            }
        }
        int[] resArr = new int[res.size()];
        int i = 0;
        for(int x : res){
            resArr[i++] = x;
        }
        return resArr;
    }
    
    private void set(int key, int val){
        if(lruMap.size() < this.k){
            lruMap.put(key, val);
        } else {
            Iterator it = lruMap.keySet().iterator();
            lruMap.remove(it.next());
            lruMap.put(key, val);
        }
    }
    
    private void get(int key){
        if(lruMap.containsKey(key)){
            int val = lruMap.get(key);
            res.add(val);
            lruMap.remove(key);
            lruMap.put(key, val);
        } else {
            res.add(-1);
        }
    }
}