LeetCode: LRU Cache

这个最坑是需求不写明。set的key如果存在的话，value要替换，同时把LRU的位置刷新。

class LRUCache{
public:

typedef list<int>::iterator ListIt;

    LRUCache(int capacity):_capacity(capacity) {
        _data.reserve(_capacity+1); 
    }
    
    int get(int key) {
        auto it = _data.find(key);                                                                                                        
        if (it==_data.end()){                                                                                                             
            return -1;                                                                                                           
        }                                                                                                                                 
        _lru.erase(it->second.second);                                                                                                    
        _lru.push_front(key);                                                                                                             
        it->second.second = _lru.begin();                                                                                                 
        return it->second.first;  
    }
    
    void set(int key, int value) {
        auto it = _data.find(key);                                                                                                        
        if(it!=_data.end()){
            it->second.first = value;
            _lru.erase(it->second.second);
            _lru.push_front(key);
            it->second.second=_lru.begin();
            return;                                                                                                                         
        }                                                                                                                       
                                                                                                                             
        if(_capacity > 0){                                                                                                                
            _capacity --;                                                                                                                   
        }else{                                                                                                                           
        int tmpKey = *(--_lru.end());                                                                                                   
        _lru.pop_back();                                                                                                                
        _data.erase(tmpKey);                                                                                                            
        }                                                                                                                                 
        _lru.push_front(key);                                                                                                             
        _data.insert(make_pair(key, make_pair(value, _lru.begin())));
    }
    list<int> _lru;
    int _capacity;
    unordered_map<int, pair<int, ListIt>> _data;
};