设计并实现最不经常使用(LFU)缓存的数据结构。它应该支持以下操作:get 和 put。
get(key) - 如果键存在于缓存中,则获取键的值(总是正数),否则返回 -1。
put(key, value) - 如果键不存在,请设置或插入值。当缓存达到其容量时,它应该在插入新项目之前,使最不经常使用的项目无效。在此问题中,当存在平局(即两个或更多个键具有相同使用频率)时,最近最少使用的键将被去除。
进阶:
你是否可以在 O(1) 时间复杂度内执行两项操作?
示例:
LFUCache cache = new LFUCache( 2 /* capacity (缓存容量) */ );
cache.put(1, 1);
cache.put(2, 2);
cache.get(1); // 返回 1
cache.put(3, 3); // 去除 key 2
cache.get(2); // 返回 -1 (未找到key 2)
cache.get(3); // 返回 3
cache.put(4, 4); // 去除 key 1
cache.get(1); // 返回 -1 (未找到 key 1)
cache.get(3); // 返回 3
cache.get(4); // 返回 4
这道题思路很简单,但是有很多细节在题目描述里并没有详细说明,比如put中已经存在key了怎么办 , 还有capacity==0的情况怎么办?(写之前需要弄清楚边界条件,切记!否则事倍功半)
put时已经存在key则更新value ;当capacity==0时不添加任何内容 ;
我设置了一个map<fre , map<time , key>> fmap,因为map是有序的,所以fmap的第一个元素就是使用频率最低的key值集合,key值集合中的第一个元素就是最近最少使用的元素 ;
class LFUCache {
public:
LFUCache(int capacity) {
cap = capacity ;
}
int get(int key) {
if(kmap.empty() || !kmap.count(key)) return -1 ;
time++ ;
int t = kmap[key].second.first , fre = kmap[key].second.second , val = kmap[key].first;
fmap[fre].erase(t) ;
if(fmap[fre].empty()) fmap.erase(fre) ;
fre++;
t = time ;
kmap[key] = {val , {t , fre}} ;
fmap[fre][t] = key ;
return val ;
}
void put(int key, int value)
{
if(kmap.count(key))
{
get(key) ;
kmap[key].first = value ;
return ;
}
time++ ;
if(kmap.size() == cap && cap > 0)
{
int fre = fmap.begin()->first , t = fmap[fre].begin()->first , k = fmap[fre][t] ;
fmap[fre].erase(t) ;
if(fmap[fre].empty()) fmap.erase(fre) ;
kmap.erase(k) ;
}
if(cap > 0)
{
kmap[key] = {value , {time , 0}} ;
fmap[0][time] = key ;
}
}
private :
int cap = 0 , time = 0 ;
unordered_map<int , pair<int , pair<int ,int>>> kmap ;//<key , <value , <t , fre>>>
map<int, map<int , int>> fmap ;
};
/**
* Your LFUCache object will be instantiated and called as such:
* LFUCache* obj = new LFUCache(capacity);
* int param_1 = obj->get(key);
* obj->put(key,value);
*/