这里k==100
#include <cstdio>
#include <string>
#include <cmath>
#include <queue>
#include <iostream>
using namespace std;
int main()
{//默认是大顶堆
priority_queue<int> q;
int n, t;
scanf("%d", &n);for(int i=0; i<n; i++)
{ scanf("%d", &t);q.push(t);
}
bool first=true;
int cnt=100;
while (!q.empty() && cnt>0)
{if(first)
{ printf("%d", q.top());first=false;
}
else
{ printf(" %d", q.top());}
q.pop();
--cnt;
}
printf("
");return 0;
}
前面算法由于queue保存了所有的数据，所以比较速度和内存占用比较大，还可以优化。
思想是，只保留最大的k个。 这样就要用最小堆，每次入队后，把最小的删除。 最后倒序输出
优化后代码如下：
#include <cstdio>
#include <string>
#include <cmath>
#include <queue>
#include <list>
#include <iostream>
using namespace std;
const int cnt=100;
int main()
{//默认是大顶堆, 改为最小堆。 小的在前
priority_queue<int, vector<int>, greater<int> > q;
int n, t;
scanf("%d", &n);for(int i=0; i<n; i++)
{ scanf("%d", &t);q.push(t);
// 超过100个，就把前面小的剔除，剩余的就是我们要的大的100个，每次插入就只要跟最大的100个比较
if(q.size()>cnt)
q.pop();
}
// 由于目前小的在前，大在后，所以需要倒序。先塞入到vector里面
vector<int> res;
while (!q.empty()) {res.push_back(q.top());
q.pop();
}
//倒序输出
for(vector<int>::reverse_iterator it = res.rbegin(); it!=res.rend(); ++it)
{if(it==res.rbegin())//first one
printf("%d", *it);else
printf(" %d", *it);}
printf("
");return 0;
}
发现速度没提高多少，但是内存小了很多。