代码改变世界

前m大的数

2019-07-18 20:15  木木王韦  阅读(115)  评论(0)    收藏  举报

**

前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

两两相加复杂度不到1e7,这里可以更加简化,因为m的范围是1000,所以只将前m个大的数进行相加排序。

#include<iostream>
#include<algorithm>
using namespace std;
int a[10000];
int b[1000100];
int m,n;
int main(){
	while(cin>>n>>m){
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	int t=0;
	sort(a,a+n);
	for(int i=n-1;i>=(n-m)&&i>=0;i--){
		//cout<<a[i]<<" ";
		for(int j=i-1;j>=(n-m)&&j>=0;j--){
			b[t++]=a[i]+a[j];
			//cout<<b[t-1]<<" ";
		}
	}
	//cout<<endl;
	sort(b,b+t);
	for(int i=t-1;i>t-m;i--){
		cout<<b[i]<<" ";
	}
	cout<<b[t-m]<<endl;
	}
	
	return 0;
}