标签:
主要是priority_queue的用法
一个是内置类型优先队列怎么设置小根堆(默认大根堆)
如果是自定义数据结构,有两种办法
1、定义这种数据结构的比较符号,就可以当成内置类型整
2、传进去一个重载()的类,当小于号用,默认还是大根堆,也许传进去的是个callable object都行的吧,我试了一下函数好像不行,不懂,不管了
LRU Cache
class LRUCache{ public: // @param capacity, an integer int Time; typedef int key; typedef int time; typedef int value; typedef pair<key,time> ktpair; struct cmp { bool operator()(ktpair x1,ktpair x2) { return x1.second > x2.second; } }; map<key,value> kv; map<key,time> kt; priority_queue<ktpair,vector<ktpair>,cmp> q; int Cap; LRUCache(int capacity) { // write your code here Time = 0; Cap = capacity; } void insert(int key,int value,int time) { kv[key] = value; kt[key] = time; q.push(make_pair(key,time)); } // @return an integer int get(int key) { // write your code here Time++; value ret; if (kv.find(key) != kv.end()) { int value = kv[key]; insert(key,value,Time); return value; } else return -1; } // @param key, an integer // @param value, an integer // @return nothing void set(int key, int value) { // write your code here Time++; if (kv.find(key) == kv.end()) { insert(key,value,Time); if (!Cap) { for(;;) { auto x = q.top(); auto ckey = x.first; auto ctime = x.second; q.pop(); if (kt.find(ckey) == kt.end() || kt[ckey] != ctime) continue; else { kv.erase(ckey); kt.erase(ckey); break; } } } else Cap--; } else insert(key,value,Time); } };
Data Stream Median
class Solution { public: /** * @param nums: A list of integers. * @return: The median of numbers */ vector<int> medianII(vector<int> &nums) { // write your code here priority_queue<int,vector<int>,less<int>> q1; priority_queue<int,vector<int>,greater<int>> q2; vector<int> ret; q1.push(INT_MIN); q2.push(INT_MAX); for (auto x : nums) { if (x < q2.top()) q1.push(x); else q2.push(x); if (q1.size() < q2.size()) { q1.push(q2.top()); q2.pop(); } if (q1.size() > 1 + q2.size()) { q2.push(q1.top()); q1.pop(); } ret.push_back(q1.top()); } return ret; } };
【Lintcode】LRU Cache, Data Stream Median
标签:
原文地址:http://www.cnblogs.com/soya/p/5205595.html