【每日一题】【迭代器，泛型】2022年1月8日-NC93 设计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)O(1)

答案：

import java.util.*;


public class Solution {
    /**
     * lru design
     * @param operators int整型二维数组 the ops
     * @param k int整型 the k
     * @return int整型一维数组
     */
    public int[] LRU (int[][] operators, int k) {
        // write code here
        List<Integer> list = new ArrayList<>();
        Map<Integer, Integer> map = new LinkedHashMap<>();
        for(int[] arr :operators) {
            if(arr[0] == 1) {
                //插入数据
                //判断map中是否包含该数据
                if(map.containsKey(arr[1])) {
                    map.remove(arr[1]);
                }
                //判断map中的容量是否已满
                if(map.size() >= k) {
                    //移除第一个元素，可以使用迭代器
                    Iterator<Integer> iterator = map.keySet().iterator();
                    int firstKey = iterator.next();
                    map.remove(firstKey);
                } 
                map.put(arr[1], arr[2]);
            } else {
                //取数据
                int num = map.getOrDefault(arr[1], -1);
                //取完数据放至头部
                if(num != -1) {
                    map.remove(arr[1]);
                    map.put(arr[1], num);
                }
                list.add(num);
            }
        }
        int[] res = new int[list.size()];
        for(int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }
}