leetCode146. LRU 缓存

leetCode146. LRU 缓存

---

题目思路

此题：1.时间复杂度o(1)  -> 对应的是哈希表
2.LRU的实现：需要记录每个key-value,而且也要记录对应的时间戳
当capacity满了后，删除时间戳最长的一个
get,put都需要更新key-value对应的时间戳（有则更新，无则插入/更换）

实现思路：用一个哈希表维护每个key-value
用一个双链表(插入，删除某一个元素都是o(1)的)，当用到某一个key,
不管哈希表是否存在都插入到首节点，并把原先的位置删除掉(如果该元素存在哈希表的话)
这样最久没用过的都会放在尾结点，当capacity满了后，把尾结点删除，把新节点插入首节点就好

---

代码

class LRUCache {
public:
    struct Node {
        int key, value;
        Node * left, * right;
        Node(int _key, int _value): key(_key), value(_value), left(nullptr), right(nullptr) {} 
    } * L, * R;

    int n = 0;
    unordered_map<int, Node *> lruCache;

    void remove(Node * p) {
        p->right->left = p->left;
        p->left->right = p->right;
    }

    void insert(Node * p) {
        p->right = L->right;
        p->left = L;
        L->right->left = p;
        L->right = p;
    }

    LRUCache(int capacity) {
        n = capacity;
        L = new Node(-1,-1), R = new Node(-1, -1);
        L->right = R;
        R->left = L;
    }
    
    int get(int key) {
        if (lruCache.count(key) == 0 ) return -1; // 表示没有找到
        auto p = lruCache[key];
        remove(p);
        insert(p);
        return p->value;
    }
    
    void put(int key, int value) {
        if (lruCache.count(key) != 0 ) {
            auto p = lruCache[key];
            remove(p);
            p->value = value; // 更新value
            insert(p);
        } else {
            if (lruCache.size() == n) {
                auto p = R->left;
                remove(p);
                // 还得把哈希表中的key和value给删除掉
                lruCache.erase(p->key);
                delete(p);
            }

            auto p = new Node(key, value);
            // 新节点得进去维护的哈希表中
            lruCache[key] = p;
            insert(p);
        }

        
    }
};

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