lru-cache.cpp

// Time:  O(1), per operation.
// Space: O(k), k is the capacity of cache.

#include <list>

class LRUCache {
public:
    LRUCache(int capacity) : capa_(capacity) {
    }
    
    int get(int key) {
        if (!map_.count(key)) {
            return -1;
        }
        // It key exists, update it.
        const auto value = map_[key]->second;
        update(key, value);
        return value;
    }
    
    void put(int key, int value) {
        if (capa_ <= 0) {
            return;
        }

        // If cache is full while inserting, remove the last one.
        if (!map_.count(key) && list_.size() == capa_) {
            auto del = list_.front(); list_.pop_front();
            map_.erase(del.first);
        }
        update(key, value);
    }
    
private:
    list<pair<int, int>> list_; // key, value
    unordered_map<int, list<pair<int, int>>::iterator> map_; // key, list iterator
    int capa_;
    
    // Update (key, iterator of (key, value)) pair
    void update(int key, int value) {
        auto it = map_.find(key);
        if (it != map_.end()) {
            list_.erase(it->second);
        }
        list_.emplace_back(key, value);
        map_[key] = prev(end(list_));
    }
};

// Time:  O(1), per operation.
// Space: O(k), k is the capacity of cache.
class LRUCache2 {
public:
    LRUCache2(int capacity) : capa_(capacity) {
    }
    
    int get(int key) {
        if (!map_.count(key)) {
            return -1;
        }
        // It key exists, update it.
        const auto value = map_[key]->second;
        update(key, value);
        return value;
    }
    
    void put(int key, int value) {
        if (capa_ <= 0) {
            return;
        }
    
        // If cache is full while inserting, remove the last one.
        if (!map_.count(key) && list_.size() == capa_) {
            auto del = list_.back(); list_.pop_back();
            map_.erase(del.first);
        }
        update(key, value);
    }
    
private:
    list<pair<int, int>> list_; // key, value
    unordered_map<int, list<pair<int, int>>::iterator> map_; // key, list iterator
    int capa_;
    
    // Update (key, iterator of (key, value)) pair
    void update(int key, int value) {
        auto it = map_.find(key);
        if (it != map_.end()) {
            list_.erase(it->second);
        }
        list_.emplace_front(key, value);
        map_[key] = list_.begin();
    }
};