146. 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lru-cache
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

基本思路：首先是考虑适合的数据结构，由于时间复杂度要求为O(1)和不断地移动链表，所以考虑使用双向链表和哈希表来存储相关的信息。

struct DLinkedNode{
    int key,val;
    DLinkedNode *prev;
    DLinkedNode *next;
    DLinkedNode():key(0),val(0),prev(nullptr),next(nullptr){};
    DLinkedNode(int _key,int _val):key(_key),val(_val),prev(nullptr),next(nullptr){};
};

class LRUCache {
private:
    DLinkedNode *head;
    DLinkedNode *tail;
    unordered_map<int,DLinkedNode*> dict;
    int size;
    int capacity;
public:
    LRUCache(int capacity) {
        head=new DLinkedNode();
        tail=new DLinkedNode();  //伪头部和伪尾部可以避免许多边界条件的讨论。
        head->next=tail;
        tail->prev=head;
        size=0;
        this->capacity=capacity;
    } 
    void moveToHead(DLinkedNode *node){
        removeNode(node);
        addToHead(node);
    }
    void removeNode(DLinkedNode *node){
        node->prev->next=node->next;
        node->next->prev=node->prev;
    }
    void addToHead(DLinkedNode *node){
        node->next=head->next;
        node->next->prev=node;
        head->next=node;
        node->prev=head;
    }
    DLinkedNode* removeFromTail(){
        DLinkedNode *node=tail->prev;
        removeNode(node);
        return node;
    }
    int get(int key) {
        if(dict.count(key)){
            DLinkedNode *cur=dict[key];
            moveToHead(cur);
            return cur->val;
        }
        else
            return -1;
    }
    
    void put(int key, int value) {
        if(dict.count(key)){
            DLinkedNode *cur=dict[key];
            cur->val=value;
            moveToHead(cur);
        }
        else{
            DLinkedNode *node=new DLinkedNode(key,value);
            addToHead(node);
            dict[key]=node;
            size++;
            if(size>capacity){
                DLinkedNode *removed=removeFromTail();
                dict.erase(removed->key);
                delete removed;  //new 对象，在不使用的时候要delete
                size--;

            }
        }
    }
};