lintcode--------前K大数II

题目描述：
实现一个数据结构，提供下面两个接口
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;
    }
};