设计LRU缓存结构

1、题目描述

链接：https://www.nowcoder.com/questionTerminal/e3769a5f49894d49b871c09cadd13a61
来源：牛客网

设计LRU缓存结构，该结构在构造时确定大小，假设大小为K，并有如下两个功能

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

[要求]

1. set和get方法的时间复杂度为O(1)
2. 某个key的set或get操作一旦发生，认为这个key的记录成了最常使用的。
3. 当缓存的大小超过K时，移除最不经常使用的记录，即set或get最久远的。

若opt=1，接下来两个整数x, y，表示set(x, y)
若opt=2，接下来一个整数x，表示get(x)，若x未出现过或已被移除，则返回-1
对于每个操作2，输出一个答案

2、结题思路

根据LinkedHashMap的特性，key是按照插入顺序存放的；

使用LinkedHashMap作为LRU缓存结构，保持最常用的数据记录总在缓存结构的后面；

（1）opt = 1：

• 缓存空间充足，将数据记录插入到LRU缓存的最后；
• 缓存空间不足，将LRU缓存头部删除，再将数据记录插入到LRU缓存的最后；

（2）opt = 2:

• 缓存中，有对应的数据，则获取对应的数据，并将对应数据删除后再插入到LRU缓存的最后；
• 缓存中，没有对应数据，获取-1；

3、编码展示

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
        LinkedHashMap<Integer,Integer> cache = new LinkedHashMap<>();
        ArrayList<Integer> list = new ArrayList<>();//结果集
        for(int[] op : operators){
            int key = op[1];
            switch(op[0]){
                case 1:
                    int value = op[2];
                    if(cache.size() < k){
                        //缓存空间足够，直接将数据插入缓存中
                        cache.put(key,value);
                    }else{
                        //缓存超出时，每次都删除缓存头
                        cache.remove(cache.keySet().iterator().next());
                        cache.put(key,value);
                    }
                    break;
                case 2:
                    if(cache.containsKey(key)){
                        int v = cache.get(key);
                        //获取到结果集
                        list.add(v);
                        //将使用过的放到缓存最后
                        cache.remove(key);
                        cache.put(key,v);
                    }else{
                        //缓存中没有，则添加-1
                        list.add(-1);
                    }
                    break;
                default:
            }
        }
        //将结果集转换为数组返回
        int[] res = new int[list.size()];
        int i = 0;
        for(int val : list){
            res[i++] = val;
        }
        return res;
    }
}