寻找大富翁
浙江桐乡乌镇共有n个人,请找出该镇上的前m个大富翁.
Input
输入包含多组测试用例.
每个用例首先包含2个整数n(0<n<=100000)和m(0<m<=10),其中: n为镇上的人数,m为需要找出的大富翁数, 接下来一行输入镇上n个人的财富值.
n和m同时为0时表示输入结束.
Output
请输出乌镇前m个大富翁的财产数,财产多的排前面,如果大富翁不足m个,则全部输出,每组输出占一行.
Sample Input
3 1
2 5 -1
5 3
1 2 3 4 5
0 0
sample output
5
5 4 3
这个题,无脑排序做是会TLE的,所以想到优先队列
先放一段刚才是写的TLE的代码
#include <bits/stdc++.h>
using namespace std;
const int N =100001;
bool cmp(int a,int b){
return a > b;
}
int main(){
int n,m;
int a[N];
while(scanf("%d%d",&n,&m)&&(n!=0&&m!=0)){
for(int i=0;i<n;i++){
cin>>a[i];
}
sort(a,a+n,cmp);
for(int i=0;i<m;i++){
if(i==m-1)
{
printf("%d\n",a[i]);
break;
}
printf("%d ",a[i]);
}
}
return 0;
}
然后是AC的优先队列的代码
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n,m,a;
while(scanf("%d%d",&n,&m)&&(n!=0&&m!=0)){
priority_queue<int> q;//构建优先队列q
for(int i=0; i<n; i++) {
scanf("%d", &a);
q.push(a); //把数据入队
}
for(int i=0; i<m; i++) {
if(q.empty()){//判断q是否已空,空则结束循环
break;
}
if(q.size()==1||i==m-1){
printf("%d\n", q.top());
}
else
printf("%d ", q.top()); //优先队列的top始终为其中最大的数据
q.pop();//出队
}
}
return 0;
}