146. LRU缓存机制 leetcode

题解

使用双向链表+hashmap存储数据,链表存储(key, value)和相对位置,hashmap存储(key, 链表节点指针)。
put使用hashmap查找key是否出现并得到元素所在节点,删去原有链表节点并在链表头部插入一个新节点,如果容量大于限制则从链表尾部删除一个元素。
get使用hashmap查找key是否出现并且删除原有节点重新插入在链表头部,实现刷新访问时间功能。
代码使用C++STL,如果要求速度可以手写链表和hashmap。

AC代码

class LRUCache {
public:
	int cap;
	list<pair<int, int>> ls;
	unordered_map<int, list<pair<int, int>>::iterator> mp;
	LRUCache(int capacity) {
		cap = capacity;
	}

	int get(int key) {
		auto mit = mp.find(key);
		if (mit != mp.end())
		{
			auto lit = mit->second;
			int val = lit->second;
			ls.erase(lit);
			ls.push_front({ key, val });
			mit->second = ls.begin();
			return val;
		}
		return -1;
	}

	void put(int key, int value) {
		auto mit = mp.find(key);
		if (mit != mp.end())
			ls.erase(mit->second);
		ls.push_front({ key, value });
		mp[key] = ls.begin();
		if (ls.size() > cap)
		{
			key = ls.back().first;
			ls.pop_back();
			mp.erase(key);
		}
	}
};

猜你喜欢

转载自blog.csdn.net/CaprYang/article/details/87870742