题目描述
设计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,输出一个答案
import java.util.*;
public class Solution {
/**
* lru design
* @param operators int整型二维数组 the ops
* @param k int整型 the k
* @return int整型一维数组
*/
// k是目前map的大小容量
public int[] LRU (int[][] operators, int k) {
// write code here
Map<Integer, Integer> map = new LinkedHashMap<>();
List<Integer> list = new LinkedList<>();
// [操作 key value]
// 看具体操作
for (int[] operator : operators) {
int key = operator[1];
switch(operator[0]) {
// 如果是1
// 那么就是set(x, y)
case 1:
// 取出value
int value = operator[2];
// 如果当前map的大小小于k,则直接加入到map中
if (map.size() < k) {
map.put(key, value); // 放入到map中
} else {
//
Iterator it = map.keySet().iterator();
map.remove(it.next()); // 去除第一个元素
map.put(key, value); // 然后将该元素加入到第一个位置
}
break;
// 如果是2
// 返回一个x=get(x),被删除或者已被移除,则返回-1
case 2:
// map包括了key
if (map.containsKey(key)) {
// 先取出这个val
int val = map.get(key);
// 将val加入到list中
list.add(val);
// 从map中移除key
map.remove(key);
// 然后将整个val放到第一个位置
map.put(key, val);
} else {
// 找不到值就返回-1
list.add(-1);
}
break;
default:
}
}
// 生成一个结果的list--复制list
int[] res = new int[list.size()];
int i = 0;
for (int val : list) {
res[i++] = val;
}
return res;
}
}
来源:牛客网
https://www.nowcoder.com/practice/e3769a5f49894d49b871c09cadd13a61?tpId=188&tqId=38300&rp=1&ru=%2Factivity%2Foj&qru=%2Fta%2Fjob-code-high-week%2Fquestion-ranking&tab=answerKey