首页 > 系统服务 > 详细

LeetCode ---LRU Cache

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

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

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!