HDU1280:前m大的数

还记得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;
}

稍后更新更多数据的情况(非暴力),哈希练习。。。

猜你喜欢

转载自blog.csdn.net/qq_41333844/article/details/81186476