题目描述
设计LRU(最近最少使用)缓存结构,该结构在构造时确定大小,假设大小为 k ,并有如下两个功能
- set(key, value):将记录(key, value)插入该结构
- 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)
示例1
输入:
[[1,1,1],[1,2,2],[1,3,2],[2,1],[1,4,4],[2,2]],3
返回值:
[1,-1]
说明:
[1,1,1],第一个1表示opt=1,要set(1,1),即将(1,1)插入缓存,缓存是{"1"=1}
[1,2,2],第一个1表示opt=1,要set(2,2),即将(2,2)插入缓存,缓存是{"1"=1,"2"=2}
[1,3,2],第一个1表示opt=1,要set(3,2),即将(3,2)插入缓存,缓存是{"1"=1,"2"=2,"3"=2}
[2,1],第一个2表示opt=2,要get(1),返回是[1],因为get(1)操作,缓存更新,缓存是{"2"=2,"3"=2,"1"=1}
[1,4,4],第一个1表示opt=1,要set(4,4),即将(4,4)插入缓存,但是缓存已经达到最大容量3,移除最不经常使用的{"2"=2},插入{"4"=4},缓存是{"3"=2,"1"=1,"4"=4}
[2,2],第一个2表示opt=2,要get(2),查找不到,返回是[1,-1]
示例2
输入:
[[1,1,1],[1,2,2],[2,1],[1,3,3],[2,2],[1,4,4],[2,1],[2,3],[2,4]],2
返回值:
[1,-1,-1,3,4]
待填空
public class Solution {
/**
* lru design
* @param operators int整型二维数组 the ops
* @param k int整型 the k
* @return int整型一维数组
*/
public int[] LRU (int[][] operators, int k) {
}
}
}
思路
LRU通俗来说,就是在容量达到上限时,不得不删除掉最不常使用的某个数据。
我们可以定义一个LRUCache类,定义一个HashMap类型的成员变量用于存储和判断数据是否存在,使用的数据结构为双向链表,并且编写set、get、刷新缓存的方法。
题解
LRUCache
package LRU;
import java.util.HashMap;
public class LRUCache {
class Node{
int key;
int value;
Node pre;
Node next;
public Node(int key,int val){
this.key = key;
this.value = val;
}
}
//头节点
Node head;
//尾节点
Node tail;
//LRU缓存的容量
int capacity;
private HashMap<Integer,Node> map;
//构造方法
public LRUCache(int capacity){
this.capacity = capacity;
map = new HashMap<>();
head = new Node(-1,-1);
tail = new Node(-1,-1);
head.next = tail;
tail.pre = head;
}
//set
public void set(int key,int value){
Node node = null;
if(map.containsKey(key)){
node = map.get(key);
node.value = value;
}else {
if(map.size() == capacity){
Node del = tail.pre;
map.remove(del.key);
delete(del);
}
node = new Node(key,value);
map.put(key,node);
}
refresh(node);
}
//get
public int get(int key){
if(map.containsKey(key)){
Node node = map.get(key);
refresh(node);
return node.value;
}
return -1;
}
//refresh
public void refresh(Node node){
delete(node);
node.next = head.next;
head.next.pre = node;
head.next = node;
node.pre = head;
}
//delete
public void delete(Node node){
if(node.pre !=null){
Node pre = node.pre;
pre.next = node.next;
node.next.pre = pre;
}
}
}
Test
package LRU;
import java.util.ArrayList;
public class Test {
public int[] LRU (int[][] operators, int k) {
ArrayList<Integer> list = new ArrayList<>();
LRUCache cache = new LRUCache(k);
for (int[] operator : operators) {
if(operator[0] == 1){
cache.set(operator[1],operator[2]);
}else if(operator[0] == 2){
list.add(cache.get(operator[1])) ;
}
}
int n = list.size();
int[] result = new int[n];
for (int i = 0; i < n; i++) {
result[i] = list.get(i);
}
return result;
}
}