问题描述:
还记得Gardon给小希布置的那个作业么?(上次比赛的1005)其实小希已经找回了原来的那张数表,现在她想确认一下她的答案是否正确,但是整个的答案是很庞大的表,小希只想让你把答案中最大的M个数告诉她就可以了。
给定一个包含N(N<=3000)个正整数的序列,每个数不超过5000,对它们两两相加得到的N*(N-1)/2个和,求出其中前M大的数(M<=1000)并按从大到小的顺序排列。
样例输入:
输入可能包含多组数据,其中每组数据包括两行:
第一行两个数N和M,
第二行N个数,表示该序列。4 4
1 2 3 4
4 5
5 3 6 4
样例输出:
对于输入的每组数据,输出M个数,表示结果。输出应当按照从大到小的顺序排列。
7 6 5 5
11 10 9 9 8
思路:由于每个数字不超过5000,所以两两相加的和小于等于10000,开一个10001长度的数组,下标表示相加的和,存储两两相加等于该下标的值的个数,然后从大到小输出计数非0的前M个值。
AC代码:
#include <cstdio>
#include <cstring>
int sort_ls[10001];
int main()
{
int N, M;
while(~scanf("%d %d", &N, &M))
{
memset(sort_ls, 0, sizeof(sort_ls));
int a[N];
for(int i = 0; i < N; i++)
{
scanf("%d", &a[i]);
}
if(M == 0)
{
continue;
}
for(int i = 0; i < N - 1; i++)
{
for(int j = i + 1; j < N; j++)
{
sort_ls[a[i] + a[j]]++;
}
}
for(int i = 10000; M > 0; i--)
{
if(sort_ls[i] == 0)
{
continue;
}
else
{
while(sort_ls[i] > 0 && M > 0)
{
if(M != 1)
{
printf("%d ", i);
sort_ls[i]--;
M--;
}
else
{
printf("%d\n", i);
sort_ls[i]--;
M--;
}
}
}
}
}
return 0;
}