还记得Gardon给小希布置的那个作业么?(上次比赛的1005)其实小希已经找回了原来的那张数表,现在她想确认一下她的答案是否正确,但是整个的答案是很庞大的表,小希只想让你把答案中最大的M个数告诉她就可以了。
给定一个包含N(N<=3000)个正整数的序列,每个数不超过5000,对它们两两相加得到的N*(N-1)/2个和,求出其中前M大的数(M<=1000)并按从大到小的顺序排列。
Input
输入可能包含多组数据,其中每组数据包括两行:
第一行两个数N和M,
第二行N个数,表示该序列。
Output
对于输入的每组数据,输出M个数,表示结果。输出应当按照从大到小的顺序排列。
Sample Input
4 4
1 2 3 4
4 5
5 3 6 4
Sample Output
7 6 5 5
11 10 9 9 8
没想到穷举没有超时。。。所以在想是否可以将原数组排序后操作。
#include<bits/stdc++.h>
using namespace std;
int a[3005],sum[3000*3000/2];
bool cmp(int a,int b)
{
return a>b;
}
int main()
{
int n,m,i,j;
while(cin>>n>>m)
{
for(i=0;i<n;i++)
cin>>a[i];
int k=0;
for(i=0;i<n-1;i++)
{
for(j=i+1;j<n;j++)
{
sum[k++]=a[i]+a[j];
}
}
sort(sum,sum+n*(n-1)/2,cmp);
cout<<sum[0];
for(i=1;i<m;i++)
{
cout<<' '<<sum[i];
}
cout<<endl;
}
return 0;
}
稍后更新更多数据的情况(非暴力),哈希练习。。。