Java编程-设计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],[1,2,2],[1,3,2],[2,1],[1,4,4],[2,2]],3
输出：[1,-1]

说明
第一次操作后：opt = 1, set("1",2),最常使用的记录为("1", 1)
第二次操作后：opt = 1, set("2",2),最常使用的记录为("2", 2)，("1", 1)变为最不常用的
第三次操作后：opt = 1, set("3",2),最常使用的记录为("3", 2)，("1", 1)还是最不常用的
第四次操作后：opt = 2, get("1"), 此时返回1，最常用的记录为("1", 1)，("2", 2)变为最不常用的
第五次操作后：opt = 1, set("4",4), 大小超过了3，所以移除此时最不常使用的记录("2", 2)，加入记录("4", 4)，
     并且为最常使用的记录，然后("3", 2)变为最不常使用的记录
第六次操作后：opt = 2, get("2"), “2”已经被移除，此时返回-1
备注:

1≤K≤N≤10^5
−2×10^9≤x,y≤2×10^9
实现代码：
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
        ArrayList<Integer> list = new ArrayList<Integer>();
        LRUCache cache = new LRUCache(k);
        for(int[] op : operators){
            if(op[0]==1){
                cache.put(op[1],op[2]);
            }else{
                int val = cache.get(op[1]);
                list.add(val);
            }
        }
        int[] ans = new int[list.size()];
        for(int i=0;i<list.size();i++){
            ans[i] = list.get(i);
        }
        return ans;
    }
}

class LRUCache extends LinkedHashMap<Integer,Integer>{
    int capacity;
     
    public LRUCache(int capacity){
        super(capacity,0.75f,true);
        this.capacity = capacity;
    }
     
    public int get(int key){
        return super.getOrDefault(key,-1);
    }
     
    public void put(int key, int val){
        super.put(key,val);
    }
     
    @Override
    protected boolean removeEldestEntry(Map.Entry<Integer,Integer> eldest){
        return size() > capacity;
    }
}