题目描述:
实现一个数据结构,提供下面两个接口
1.add(number) 添加一个元素
2.topk() 返回前K大的数
样例:
s = new Solution(3);
create a new data structure.
s.add(3)
s.add(10)
s.topk()
return [10, 3]
s.add(1000)
s.add(-99)
s.topk()
return [1000, 10, 3]
s.add(4)
s.topk()
return [1000, 10, 4]
s.add(100)
s.topk()
return [1000, 100, 10]
思路如前k大的数一样,只是这里需要注意下,每次调用topk的时候,由于我们需要将优先队列里的拿出来,就会导致下一次调用的时候,优先队列中已经没有上次的数了,所以我们再每次调用topk之后,都需要重新将拿出来的数添加进入。
代码详解:
class Solution {
public:
/*
* @param k: An integer
*/
int count;
priority_queue<int>q;
Solution(int k) {
// do intialization if necessary
count=k;
}
/*
* @param num: Number to be added
* @return: nothing
*/
void add(int num) {
// write your code here
q.push(num);
}
/*
* @return: Top k element
*/
vector<int> topk() {
// write your code here
vector<int>res;
int len=q.size();
if(len>=count){
int i=0;
while(i!=count){
res.push_back(q.top());
q.pop();
i++;
}
for(int i=0;i<res.size();i++){
q.push(res[i]);
}
}else{
while(!q.empty()){
res.push_back(q.top());
q.pop();
}
for(int i=0;i<res.size();i++){
q.push(res[i]);
}
}
return res;
}
};