LeetCode: LRU Cache

思路:借鉴了网上的讨论,用一个unordered_map和list组合数据结构,unordered_map存储关键字是否存在,list存储相应的值,最近使用的放在头部,否则放在尾部,unordered_map的第二项存储值在list中的位置,即迭代器。然后知道LRU的原理,get(key)需要将key对应的值放在list的头部,m[key]要做相应的变换,这个刚开始没有写上,挂了好多次。后面的set(key,value)反到是容易很多,考虑已经存储的元素大小size和容量capacity的关系,size<capacity时直接插入list,否则删除list末尾的元素再来插入key到list头部,当然,每次更新list一定要记得对map[key]值做相应的变换,以便我们知道key对应的value存在于list的何处。

class LRUCache{
public:
    LRUCache(int capacity) {
        cap = capacity;
		size = 0;
    }
    
    int get(int key) {
		if(cap<1)return -1;
		unordered_map<int,list<int>::iterator>::iterator it;
		it = m.find(key);
		if(it != m.end()){
			list<int>::iterator list_it = it->second;
			int ret = *list_it;
			L.erase(list_it);
			L.push_front(ret);
			m[key] = L.begin();
			return ret;
		}
		else
			return -1;
    }
    
    void set(int key, int value) {
		if(cap<1)return;
        unordered_map<int,list<int>::iterator>::iterator it;
		it = m.find(key);
		if(it!=m.end()){
			list<int>::iterator list_it = it->second;
			L.erase(list_it);
			L.push_front(value);
			m[key] = L.begin();
		}
		else{
			if(size <cap){
				L.push_front(value);
				m[key] = L.begin();
				size++;
			}
			else{
				list<int>::iterator list_it2 = L.end();
				list_it2--;
				unordered_map<int,list<int>::iterator>::iterator it2;
				for(it2= m.begin();it2!=m.end();it2++)
					if(it2->second == list_it2) break;
				//cout<<"hello"<<endl;
				if(it2!=m.end()){
				m.erase(it2);
				L.pop_back();
				L.push_front(value);
				m[key] = L.begin();
				}
			}
			
		}
		
    }
	void output(){
		cout<<cap<<" "<<size<<endl;
		cout<<*L.begin()<<endl;
	}
private:
	int cap;
	int size;
	list<int> L;   //存储value
	unordered_map<int,list<int>::iterator> m;  //存储关键字在list中的位置
};

其中有一个小疑问:在

if(it2!=m.end()){
				m.erase(it2);
				L.pop_back();
				L.push_front(value);
				m[key] = L.begin();
				}
如果去掉if判断,直接去操作,会出现RE,不知怎么回事,回头再想。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值