leetcode146. LRU缓存机制

146. LRU缓存机制

运用你所掌握的数据结构，设计和实现一个 LRU (最近最少使用) 缓存机制。它应该支持以下操作： 获取数据 get 和 写入数据 put 。

获取数据 get(key) - 如果关键字 (key) 存在于缓存中，则获取关键字的值（总是正数），否则返回 -1。
写入数据 put(key, value) - 如果关键字已经存在，则变更其数据值；如果关键字不存在，则插入该组「关键字/值」。当缓存容量达到上限时，它应该在写入新数据之前删除最久未使用的数据值，从而为新的数据值留出空间。

进阶:

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

示例:

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

解题思路

• 哈希表unordererd_map+双向链表实现

//LRU缓存机制
//哈希表unordererd_map+双向链表实现
struct DLinkList {
    int key;
    int value;
    DLinkList* prev;//前驱
    DLinkList* next;//后继
    DLinkList():key(0),value(0),prev(NULL),next(NULL){}//默认构造函数
    DLinkList(int _key, int _value){
        key = _key;
        value = _value;
    }
};

class LRUCache {
private:
    unordered_map<int,DLinkList*> mp;//哈希表，存key在链表中的结点
    DLinkList* DummyHead;//虚拟头结点
    DLinkList* DummyRail;//虚拟尾结点
    int curCapacity;//当前容量
    int capacity;//总容量
public:
    //构造函数
    LRUCache(int capacity) {
        this->capacity = capacity;
        curCapacity = 0;

        //初始化双向链表
        DummyHead = new DLinkList();
        DummyRail = new DLinkList();
        DummyHead->next = DummyRail;
        DummyRail->prev = DummyHead;
    }

    int get(int key) {
        //判断key是否存在
        if(mp[key]){
            //存在：将key所在的结点移到链头
            DLinkList* tmp = mp[key];
            tmp->prev->next = tmp->next;
            tmp->next->prev = tmp->prev;

            tmp->prev = DummyHead;
            tmp->next = DummyHead->next;
            DummyHead->next = tmp;
            tmp->next->prev = tmp;

            return tmp->value;
        }
        //不存在：返回-1
        else
            return -1;
    }

    void put(int key, int value) {
        //判断key是否存在
        //已存在
        if(mp[key]){
            //修改对应value值
            DLinkList* tmp = mp[key];
            tmp->key = key;
            tmp->value = value;
            //移到链头
            tmp->prev->next = tmp->next;
            tmp->next->prev = tmp->prev;

            tmp->prev = DummyHead;
            tmp->next = DummyHead->next;
            DummyHead->next = tmp;
            tmp->next->prev = tmp;
        }
        //不存在
        else {
            //判断容量是否满了
            if(curCapacity==capacity){
                //删除链尾结点（最近最久未使用）
                DLinkList* tmp = DummyRail->prev;
                DummyRail->prev = tmp->prev;
                DummyRail->prev->next = DummyRail;
                //哈希表中也删除该结点*
                mp.erase(tmp->key);
                //内存中删除该节点*
                delete tmp;
                //当前容量-1
                curCapacity--;
            }
            //链头插入key
            DLinkList* tmp = new DLinkList(key,value);
            tmp->next = DummyHead->next;
            tmp->prev = DummyHead;
            DummyHead->next = tmp;
            tmp->next->prev = tmp;
            //将key加入哈希表
            mp[key] = tmp;
            //当前容量+1
            curCapacity++;
        }
    }
};

执行用时：188 ms, 在所有 C++ 提交中击败了87.41% 的用户
内存消耗：37.3 MB, 在所有 C++ 提交中击败了60.12% 的用户