460. LFU 缓存

· 约 5 分钟阅读 · 次阅读 leetcode

请你为 最不经常使用(LFU)缓存算法设计并实现数据结构。

实现 LFUCache 类:

LFUCache(int capacity) - 用数据结构的容量 capacity 初始化对象 int get(int key) - 如果键 key 存在于缓存中,则获取键的值,否则返回 -1 。 void put(int key, int value) - 如果键 key 已存在,则变更其值;如果键不存在,请插入键值对。当缓存达到其容量 capacity 时,则应该在插入新项之前,移除最不经常使用的项。在此问题中,当存在平局(即两个或更多个键具有相同使用频率)时,应该去除 最近最久未使用 的键。 为了确定最不常使用的键,可以为缓存中的每个键维护一个 使用计数器 。使用计数最小的键是最久未使用的键。

当一个键首次插入到缓存中时,它的使用计数器被设置为 1 (由于 put 操作)。对缓存中的键执行 get 或 put 操作,使用计数器的值将会递增。

函数 get 和 put 必须以 O(1) 的平均时间复杂度运行。

 

示例:

输入: [“LFUCache”, “put”, “put”, “get”, “put”, “get”, “get”, “put”, “get”, “get”, “get”] [[2], [1, 1], [2, 2], [1], [3, 3], [2], [3], [4, 4], [1], [3], [4]] 输出: [null, null, null, 1, null, -1, 3, null, -1, 3, 4]

解释: // cnt(x) = 键 x 的使用计数

// cache=[] 将显示最后一次使用的顺序(最左边的元素是最近的) LFUCache lfu = new LFUCache(2); lfu.put(1, 1); // cache=[1,_], cnt(1)=1 lfu.put(2, 2);

// cache=[2,1], cnt(2)=1, cnt(1)=1 lfu.get(1);

// 返回 1

// cache=[1,2], cnt(2)=1, cnt(1)=2 lfu.put(3, 3);

// 去除键 2 ,因为 cnt(2)=1 ,使用计数最小

// cache=[3,1], cnt(3)=1, cnt(1)=2 lfu.get(2);

// 返回 -1(未找到) lfu.get(3);

// 返回 3

// cache=[3,1], cnt(3)=2, cnt(1)=2 lfu.put(4, 4);

// 去除键 1 ,1 和 3 的 cnt 相同,但 1 最久未使用

// cache=[4,3], cnt(4)=1, cnt(3)=2 lfu.get(1);

// 返回 -1(未找到) lfu.get(3);

// 返回 3

// cache=[3,4], cnt(4)=1, cnt(3)=3 lfu.get(4);

// 返回 4

// cache=[3,4], cnt(4)=2, cnt(3)=3

提示:

0 <= capacity <= 104 0 <= key <= 105 0 <= value <= 109 最多调用 2 * 105 次 get 和 put 方法


使用哈希表+平衡二叉树

最好使用堆,但是堆支持动态更新非头节点,使用平衡二叉树替代

时间复杂度O(logn) 空间复杂度O(capacity)

struct Node{
    int key, value, time, cnt;
    Node(int _key, int _value, int _time, int _cnt):key(_key),value(_value),time(_time),cnt(_cnt){}
    bool operator< (const Node& node) const{
        return cnt == node.cnt ? time < node.time : cnt < node.cnt; 
    }
};

class LFUCache {
    // cache容量和时间戳
    int capacity, timestamp;
    // 平衡二叉查找树,红黑树
    set<Node> S;
    // 无序map
    unordered_map<int, Node> hash_table;
public:
    LFUCache(int _capacity) {
        capacity = _capacity;
        S.clear();
        hash_table.clear();
        // 当前时间戳初始化为0
        timestamp = 0;
    }
    
    int get(int key) {
        if (capacity == 0) return -1;
        auto it = hash_table.find(key);
        if (it == hash_table.end()) return -1;
        Node cache = it->second;
        S.erase(cache);
        cache.time = ++timestamp;
        cache.cnt += 1;
        S.insert(cache);
        it->second = cache;
        return cache.value;
    }
    
    void put(int key, int value) {
        if (capacity == 0) return ;
        auto it = hash_table.find(key);
        // 未找到相同元素,新增
        if (it == hash_table.end()){
            // 存储已满,需要剔除频次最小,时间最久的一个元素
            if (hash_table.size() == capacity){
                hash_table.erase(S.begin()->key);
                S.erase(S.begin());
            }
            Node cache = Node(key, value, ++timestamp, 1);
            S.insert(cache);
            hash_table.insert(make_pair(key, cache));
        }else{ //更新
            Node cache = it->second;
            S.erase(cache);
            cache.time = ++timestamp;
            cache.cnt += 1;
            cache.value = value;
            S.insert(cache);
            it->second = cache;
        }


    }
};

/**
 * 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);
 */

两个哈希表

一个哈希表frequency_table存储频率和对应频率下的节点,使用双向链表解决冲突, 队头的节点是最近的节点,队尾的节点是较久的节点。 每个节点存储,key,value,freq

一个哈希表key_table存储key和对应缓存在双向链表中的地址。

时间复杂度变为O(1)

struct Node{
    int key, value, freq;
    Node(int _key, int _value, int _freq):key(_key),value(_value),freq(_freq){}
};

class LFUCache {
    int minfreq, capacity;
    unordered_map<int, list<Node>> freq_table;
    unordered_map<int, list<Node>::iterator> key_table;
public:
    LFUCache(int _capacity) {
        capacity = _capacity;
        minfreq = 0;
        freq_table.clear();
        key_table.clear();
    }
    
    int get(int key) {
        if (capacity == 0) return -1;
        auto it = key_table.find(key);
        if (it == key_table.end()) return -1;
        list<Node>::iterator node = it->second;
        int freq = node->freq;
        int value = node->value;
        //需要更新频率
        freq_table[freq].erase(node);
        if (freq_table[freq].size() == 0){
            freq_table.erase(freq);
            if (freq == minfreq) minfreq += 1;
        }
        freq_table[freq+1].push_front(Node(key, value, freq+1));
        key_table[key] = freq_table[freq+1].begin();
        return value;
    }
    
    void put(int key, int value) {
        if (capacity == 0) return ;
        auto it = key_table.find(key);
        // 未找到相同元素,新增
        if (it == key_table.end()){
            // 存储已满,需要剔除频次最小,时间最久的一个元素
            if (key_table.size() == capacity){
                // back返回的是最后一个元素的引用
                auto it2 = freq_table[minfreq].back();
                key_table.erase(it2.key);
                freq_table[minfreq].pop_back();
                if (freq_table[minfreq].size() == 0){
                    freq_table.erase(minfreq);
                }
            }
            freq_table[1].push_front(Node(key, value, 1));
            key_table[key] = freq_table[1].begin();
            minfreq = 1;
        }else{ //更新
            // iterator相当于指针,也表示内存地址,*node则为一个Node对象。访问对象成员变量用, 指针访问变量用->
            list<Node>::iterator node = it->second;
            int freq = node->freq;
            freq_table[freq].erase(node);
            if (freq_table[freq].size() == 0){
                freq_table.erase(freq);
                if (freq == minfreq) minfreq+=1;
            }
            freq_table[freq+1].push_front(Node(key, value, freq+1));
            key_table[key] = freq_table[freq+1].begin();
        }


    }
};

/**
 * 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);
 */