题目描述;
在一个数组中找到前K大的数
样例:
给出 [3,10,1000,-99,4,100], k = 3.
返回 [1000, 100, 10]
思路讲解:这里我的第一想法就是利用一个小的res,首先将里面添加进nums的前k个数,然后对其排序,之后我们就将nums位置k后面的数逐个的与res数组比较这里就用到了对有序数组的插入排序,结果这样很不辛的在70%的时候超时了,后面我看了一下标签,发现可以用优先队列,首先我们将所有的数,都进入优先队列,然后将其前k拿出来就好了。后面想了一下,我之前的那种算法适用的场景是k很小,并且数据很大的时候。
代码详解:
class Solution {
public:
/**
* @param nums: an integer array
* @param k: An integer
* @return: the top k largest numbers in array
*/
vector<int> topk(vector<int> &nums, int k) {
// write your code here
int length=nums.size();
priority_queue<int> q;
vector<int>res;
for(int i=0;i<length;i++){
q.push(nums[i]);
}
int count=0;
while(count!=k){
res.push_back(q.top());
q.pop();
count++;
}
printvector(res);
return res;
}
void printvector(vector<int>res){
//cout<<"length is "<<res.size()<<endl;
for(int i=0;i<res.size();i++){
cout<<res[i]<<" ";
}
cout<<endl;
}
};