【题解】LRU缓存

题目

题目来源：leetcode 面试题 16.25 LRU缓存

设计和构建一个“最近最少使用”缓存，该缓存会删除最近最少使用的项目。缓存应该从键映射到值(允许你插入和检索特定键对应的值)，并在初始化时指定最大容量。当缓存被填满时，它应该删除最近最少使用的项目。

它应该支持以下操作： 获取数据 get 和 写入数据 put 。

获取数据 get(key)：如果密钥 (key) 存在于缓存中，则获取密钥的值（总是正数），否则返回 -1。
写入数据 put(key, value)：如果密钥不存在，则写入其数据值。当缓存容量达到上限时，它应该在写入新数据之前删除最近最少使用的数据值，从而为新的数据值留出空间。
例：

LRUCache cache = new LRUCache( 2 /* 缓存容量 */ );
cache.put(1, 1);
cache.put(2, 2);
cache.get(1); // 返回 1
cache.put(3, 3); // 该操作会使得密钥 2 作废
cache.get(2); // 返回 -1 (未找到)
cache.put(4, 4); // 该操作会使得密钥 1 作废
cache.get(1); // 返回 -1 (未找到)
cache.get(3); // 返回 3
cache.get(4); // 返回 4

解题思路

本文使用双向链表构建LRU缓存，链表中每个节点表示一块缓存（包含key和value）。除此之外，还用一个哈希表将缓存的key值和对应链表节点的地址关联起来。

每次访问一个节点时，都要将它移动到链表的第一个位置上，并更新哈希表中对应的项。
每次删除一个节点时，要删除的是链表中最后一个位置的节点，同时也要删除哈希表中对应的项。
每次添加一个节点时，要将新建的节点添加在链表第一个位置，并向哈希表中添加对应的项。

缓存工作的大体流程是：
（1） 初始化时，指定缓存的最大容量。

（2） 执行get时，通过key值在哈希表上找相应节点的地址：a.如果找不到该地址，则返回-1；b.否则将该节点移动到链表的第一个位置，并更新哈希表中对应的项，返回该节点的value字段。

（3） 执行put时，通过key值在哈希表上找相应节点的地址：a.如果找不到该地址且缓存已满，则删除链表中最后一个位置的节点，也把哈希表中的对应项删除掉（若缓存未满，就不用删除节点和哈希表中的对应项）。以新的key和value值在链表的第一个位置创建节点，在哈希表中也增添相应的项。b.否则更新链表中相应节点的value字段，并将该节点移动到链表的第一个位置，再更新哈希表中对应的项。

代码实现（C++）

class LRUCache {
public:
    LRUCache(int capacity) {
        this->capacity = capacity;
    }
    
    int get(int key) {
        auto it = (this->um).find(key);
        if (it != (this->um).end()) {
            Node tmp = *(it->second);
            (this->ln).erase(it->second);
            (this->ln).push_front(tmp);
            (this->um)[key] = (this->ln).begin();
            return (this->ln).front().value;
        }
        else {
            return -1;
        }
    }
    
    void put(int key, int value) {
        auto it = (this->um).find(key);
        if (it == (this->um).end()) {
            if ((this->ln).size() == this->capacity) {
                (this->um).erase((this->ln).back().key);
                (this->ln).pop_back();
            }
            (this->ln).push_front(Node(key, value));
            (this->um)[key] = (this->ln).begin();
        }
        else {
            it->second->value = value;
            Node tmp = *(it->second);
            (this->ln).erase(it->second);
            (this->ln).push_front(tmp);
            (this->um)[key] = (this->ln).begin();
        }
    }
private:
    class Node {
    public:
        int key;
        int value;
        Node(int k, int v) : key(k), value(v) {}
    };
    list<Node> ln;
    int capacity;
    unordered_map<int, list<Node>::iterator> um;
};

/**
 * Your LRUCache object will be instantiated and called as such:
 * LRUCache* obj = new LRUCache(capacity);
 * int param_1 = obj->get(key);
 * obj->put(key,value);
 */