LeetCode: LRU Cache

题目

https://oj.leetcode.com/problems/lru-cache/

分析

这道题主要考察的是数据结构的设计能力。

首先考虑用hash_map来存储数据，这样get()可以达到O(1)；不过hash_map没有淘汰LRU Cache的高效办法：一种方法是每个node存放一个最近使用的时间，每次set()时遍历时找出LRU node，这样set()是O(n)的时间复杂度，而且这样设计的话每次还要计算出哪个节点和现在的时间差值最大，实现起来特别麻烦。

之前面试的时候面试官提示可以用list实现：每次把最近用到的节点放到list最前面。但是当时觉得这种方法不太好，因为get()和set()的时间复杂度都是O(n)，效率实在太低。

现在来看，应该是把hash_map和链表结合起来来用，使得get()和set()都能够达到O(1)的时间复杂度：每次用到的数据添加到list最前面，每次需要淘汰数据时，淘汰list的最后面的元素。同时用hash_map记录每个key在list中的位置，这样每次要根据key查询数据时，便到hash_map中查找数据。

代码

class LRUCache{public:LRUCache(int capacity){capacity_ = capacity;}int get(int key){if (cache_map_.find(key) == cache_map_.end())return -1;else{//delete old node and push a new node in front of listint temp = cache_map_[key]->second;cache_.erase(cache_map_[key]);cache_.push_front(pair<int, int>(key, temp));cache_map_[key] = cache_.begin();return temp;}}void set(int key, int value){if (cache_map_.find(key) == cache_map_.end()){if (cache_.size() >= capacity_){//delete the last node of listcache_map_.erase(cache_.back().first);cache_.pop_back();}//push new node in front of listcache_.push_front(pair<int, int>(key, value));cache_map_[key] = cache_.begin();}else{//delete old node and push a new node in front of  listcache_.erase(cache_map_[key]);cache_.push_front(pair<int, int>(key, value));cache_map_[key] = cache_.begin();}}private:int capacity_;list<pair<int, int>> cache_;unordered_map<int, list<pair<int, int>>::iterator> cache_map_;};

Note

在写代码的过程中，犯了很多2B的错误，导致花了很久才找出错误。

1. 在更新list中的node时，忘记了同时更新map中的数据。

2. 错误的把list.end()当做了list最后一个节点。 list.back()才是list最后一个节点。

参考

http://fisherlei.blogspot.com/2013/11/leetcode-lru-cache-solution.html