码迷,mamicode.com
首页 > 系统相关 > 详细

LeetCode ---LRU Cache

时间:2015-07-24 18:31:55      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:lru-cache

Design and implement a data structure for Least Recently Used (LRU) cache. It should support the following operations: get and set.

  • get(key) - Get the value (will always be positive) of the key if the
    key exists in the cache, otherwise return -1.

  • set(key, value) - Set or insert the value if the key is not already present. When the cache reached its capacity, it should invalidate the least recently used item before inserting a new item.

class LRUCache{
public:
    LRUCache(int capacity) 
    {
         m_cache.clear();
         m_time2key.clear();
         m_key2time.clear();
         m_capacity = capacity;
         m_time = 0;
         m_size = 0;
    }

    int get(int key)
    {
        m_time ++;
        map<int, int >::iterator it = m_cache.find(key);
        if(it != m_cache.end() )
        {
            m_time2key.erase(m_key2time[key]);
            m_time2key.insert(std::make_pair(m_time, key));
            m_key2time[key] = m_time;
            return m_cache[key];
        }

        return -1;
    }

    void set(int key, int value) 
    {
        m_time++;

        map<int, int >::iterator it = m_cache.find(key);
        if(it != m_cache.end())
        {
            m_cache[key] = value;
            m_time2key.insert(std::make_pair(m_time, key));
            m_time2key.erase(m_key2time[key]);
            m_key2time[key] = m_time;
        }
        else
        {
            if(m_size == m_capacity)
            {
                map<long long, int>::iterator it = m_time2key.begin();
                m_cache.erase(it->second);
                m_time2key.erase(it->first);
                m_key2time.erase(it->second);
                m_size--;
            }
            m_cache.insert(std::make_pair(key, value));
            m_time2key.insert(std::make_pair(m_time, key));
            m_key2time.insert(std::make_pair(key, m_time));
            m_size++;
        }
    }
private:
    map<int, int > m_cache;
    long long m_time;
    map<long long, int> m_time2key;
    map<int, long long> m_key2time;
    int m_capacity;
    int m_size;
};

版权声明:本文为博主原创文章,未经博主允许不得转载。

LeetCode ---LRU Cache

标签:lru-cache

原文地址:http://blog.csdn.net/ithiker/article/details/47042871

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!