【HOT 100】146. LRU 缓存机制

146. LRU 缓存机制

题目描述

运用你所掌握的数据结构，设计和实现一个  LRU (最近最少使用) 缓存机制 。
实现 LRUCache 类：

• LRUCache(int capacity) 以正整数作为容量 capacity 初始化 LRU 缓存
• int get(int key) 如果关键字 key 存在于缓存中，则返回关键字的值，否则返回 -1 。
• void put(int key, int value) 如果关键字已经存在，则变更其数据值；如果关键字不存在，则插入该组「关键字-值」。当缓存容量达到上限时，它应该在写入新数据之前删除最久未使用的数据值，从而为新的数据值留出空间。

进阶：你是否可以在 O(1) 时间复杂度内完成这两种操作？

方法一：哈希表 + 双向链表

LRU（Least Recently Used），最近最少使用算法，是一种常用的页面置换算法，选择最近最久未使用的页面予以淘汰。

使用哈希表来实现 O(1) 时间复杂度的 get，再辅以双向链表实现 O(1) 时间复杂的 put，同时双向链表还实现了对数据按照访问时间排序。具体来说：

• 双向链表结点包含对 key,value 的封装
• 哈希表以数据的 <key，DLinkedNode> 为映射，通过 key 映射到其在双向链表中的结点。

// 双向链表
struct DLinkedNode
{
    int key;
    int value;
    DLinkedNode *prev;
    DLinkedNode *next;
    DLinkedNode() : key(0), value(0), prev(nullptr), next(nullptr){};
    DLinkedNode(int key, int value) : key(key), value(value), prev(nullptr), next(nullptr){};
};

class LRUCache
{
private:
    unordered_map<int, DLinkedNode *> cache;
    DLinkedNode *head; // 双向链表头指针
    DLinkedNode *tail; // 双向链表尾指针
    int capacity;      // cache 容量
    int size;          // 双向链表的大小

public:
    LRUCache(int capacity) : capacity(capacity), size(0)
    {
        // 初始化的时候需要先创建头尾结点，并且头尾指针要构成链
        head = new DLinkedNode();
        tail = new DLinkedNode();
        head->next = tail;
        tail->prev = head;
    }
    /*
      get 操作，判断 key 是否存在
        1. 不存在，返回 -1
        2. 存在，将 key 对应的结点移至双向链表表头，返回该结点的值
    */
    int get(int key)
    {
        if (!cache.count(key))
            return -1;
        DLinkedNode *node = cache[key];
        moveToHead(node);
        return node->value;
    }

    /*
      put 操作，判断 key 是否存在
        1. 不存在
            1）创建 key,value 的新结点，将此 <key,DLinkedNode> 加入到哈希表映射中。
            2) 判断缓存是否已满，若满了，需要删除最后一个结点，并在哈希表中删除映射。
            3）在双向链表头插入此结点。
        2. 存在
            1) 将 key 对应结点移至表头
            2) 更新结点的 value 值
    */
    void put(int key, int value)
    {
        if (!cache.count(key)) // 不存在
        {
            DLinkedNode *node = new DLinkedNode(key, value);
            cache[key] = node;
            if (capacity == size)
            {
                DLinkedNode *node = removeTail();
                cache.erase(node->key);
                delete node;
            }
            addToHead(node);
        }
        else
        {
            DLinkedNode *node = cache[key];
            node->value = value;
            moveToHead(node);
        }
    }

    void moveToHead(DLinkedNode *node)
    {
        // 当前节点脱链
        node->prev->next = node->next;
        node->next->prev = node->prev;
        size--;
        addToHead(node);
    }

    // 将尾结点从链表脱离，并返回尾结点
    DLinkedNode *removeTail()
    {
        DLinkedNode *node = tail->prev;
        node->prev->next = tail;
        tail->prev = node->prev;
        size--;
        return node;
    }

    void addToHead(DLinkedNode *node)
    {
        // 将当前节点移至链表头
        node->next = head->next;
        node->prev = head;
        head->next->prev = node;
        head->next = node;
        size++;
    }
};

int main()
{
    // 验证
    LRUCache my_lru(2);
    my_lru.put(1, 1);
    my_lru.put(2, 2);
    my_lru.get(1);
    my_lru.put(3, 3);
    my_lru.get(2);
    my_lru.put(4, 4);
    my_lru.get(1);
    my_lru.get(3);
    my_lru.get(4);
}