注意:需要用到unordered_map和list两个数据结构,其中需要用到的方法有list.remove(int val),可以直接移除list中值为val的所有元素。而list.erase(iterator)是迭代器的精准删除。unordered_map.erase(int key)用键值删除。
描述
设计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)O(1)
class Solution {
private:
unordered_map<int,int>ump;
list<int> ls;
int capacity;
int get(int key){
auto it=ump.find(key);
if(it!=ump.end()){//如果命中了
//ls中删除旧的,然后放到第一个
ls.remove(key);
ls.push_front(key);
return it->second;
}
else{
return -1;
}
}
void set(int key,int value){
auto it=ump.find(key);
//只有当key值未命中,且超出capacity的时候需要对LRU最久未使用的元素移除
if(it==ump.end()&&ump.size()>=capacity){
int tmp_key=ls.back();
ls.pop_back();
ump.erase(tmp_key);
}
ump[key]=value;
ls.push_front(key);
}
public:
/**
* lru design
* @param operators int整型vector<vector<>> the ops
* @param k int整型 the k
* @return int整型vector
*/
vector<int> LRU(vector<vector<int> >& operators, int k) {
// write code here
//用两个数据结构,一个是unordered_map<int,pair<int,int>>ump,用来搜索缓存
//一个是list,用来实现LRU最不常用的在末端,最常用的在首端
vector<int> res;
capacity=k;
for(int i=0;i<operators.size();i++){
if(operators[i][0]==1) set(operators[i][1],operators[i][2]);
else res.push_back(get(operators[i][1]));
}
return res;
}
};