LeetCode 146 LRU缓存机制

题目描述

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

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

进阶:

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

分析

LRU是一种常见的页面置换算法，淘汰最近最久没有使用的页面。本题要求我们在O(1)的时间复杂度内实现 get 和 put两种操作。
get操作是获取某一个key对应的value，是一个查询操作，可以想到hash表能够在常数复杂度内实现查询；
put操作是插入一对(key, value)，如果key已经存在，就更新对应的value，如果没有就插入。要注意put操作后，key对应的缓存已经变成最新使用的了，此时如果超出了容量限制，就要淘汰最久未使用的，所以put操作包含 更新/插入、调整顺序、超出容量淘汰三个部分。

首先我们知道，如果只使用一个哈希表，能够做到查询、插入、删除是常数时间，但是由于没有顺序，我们不知道淘汰哪一个，因此还需要一个数据结构记录缓存被使用的顺序。如果这一结构采用顺序存储比如vector，那在调整顺序的时候就会出现一部分存储空间整体移动的情况，这不符合时间复杂度要求，因此考虑采用链表的形式存储。

所以，首先我们设计两个数据结构如下：

list<pair<int,int>>l;
unordered_map<int,list<pair<int,int>>::iterator>m;

这里list每个节点保存了一个pair，里面分别是key和value，list的从头到尾的顺序表示从最新使用缓存到最久未用的缓存顺序，也就是说，最新使用一个缓存，就放在l的begin()位置，要淘汰就从末尾开始淘汰。unordered_map则保存key和对应的节点在list中的迭代器（指针）。

在get操作时，通过哈希表find操作查找key能够得到一个unordered_map的迭代器it，那么如果it是m.end()表示没找到，也就是这个key不存在，根据题意返回-1。反之如果存在，那么it->second就是对应缓存保存在list的迭代器，其中it->second->first是key，it->second->second是value，返回value即可。要注意由于key对应的缓存被访问，他应该变成最新的了，应该放在l.begin()的位置，这里用list::splice方法在常数时间内实现（实际上就是调整链表节点指针顺序，不涉及内存拷贝）。显然在常数时间复杂度内能够实现get操作。

put操作，我们首先还是查找m中有没有key，有的话就找到list对应节点删掉，同样的，链表删除只是调整指针并释放内存，是常数时间。然后在list的开始位置，重新插入{key,value}，然后更新哈希表记录的key对应的list迭代器信息。这样插入操作就完成了，并且最新插入的也放在了起始位置，然后就是检查是否超出缓存需要淘汰的问题。在淘汰时，我们是从链表最后开始删除，这里使用了反向迭代器rbegin()，找到最后一个节点对应的key并从哈希表移除，将最后一个节点弹出。

stl容器通过迭代器的遍历，常用的正向的方法，c.begin()表示指向第一个元素的迭代器，c.end()表示指向最后一个元素后面一个位置的迭代器 ；类似的反向迭代器 c.rbegin()表示指向最后一个元素的迭代器，而 c.rend()表示指向第一个元素前一个位置的迭代器。

C++代码如下

class LRUCache {
public:
    LRUCache(int capacity) {
        cap=capacity;
    }
    
    int get(int key) {
        auto it = m.find(key);
        if (it == m.end()) return -1;
        l.splice(l.begin(), l, it->second);//put it->second at l.begin()
        return it->second->second;
    }
    
    void put(int key, int value) {
	 	auto it = m.find(key);
	    if (it != m.end()) l.erase(it->second);
	    l.push_front({ key,value });
	    m[key] = l.begin();
	    if (m.size() > cap) {
	      int k = l.rbegin()->first;//反向迭代器
	      l.pop_back();
	      m.erase(k);
	    }
    }
    list<pair<int,int>>l;
    unordered_map<int,list<pair<int,int>>::iterator>m;
    int cap=0;
};