1、题目描述
链接:https://www.nowcoder.com/questionTerminal/e3769a5f49894d49b871c09cadd13a61
来源:牛客网
设计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,输出一个答案
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;
}
}