题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1280
题目大意:给n个数,求出所有两两相加的数,然后输出这些数中的前m大个(从大到小排列)
解题思路:使用小数优先的有限队列,维持队列的元素数量始终等于m,每当读取到一个数比队头大时就让这个数入队,同时将队头元素出队,保证元素个数为m,将所有数字扫描一遍过后即得到前m大个数,但是由于在队列中是从小到大排列的,需要将所有元素取出后反向输出
AC代码:
#include <iostream>
#include <queue>
using namespace std;
struct comp2//降序排列
{
bool operator ()(int & a, int & b)
{
return a>b;
}
};
priority_queue<int,vector<int>,comp2>q2;
int main()
{
int n,m;
int data[3005];
int result[1005];
int index=0;
while(cin>>n>>m)
{
q2.push(-1);
for(int i=0;i<n;i++)
{
cin>>data[i];
}
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
int temp = data[i]+data[j];
if(temp>q2.top())
{
q2.push(temp);
if(q2.size()>m)q2.pop();
//cout<<q2.top()<<endl;
}
}
}
index=0;
while(!q2.empty())
{
result[index++] = q2.top();
q2.pop();
}
for(int i=index-1;i>=0;i--)
{
if(i==index-1)cout<<result[i];
else cout<<" "<<result[i];
}
cout<<endl;
}
return 0;
}