HDU 1280 前m大的数(计数排序)

来源:互联网 发布:二战日本的海军数据 编辑:程序博客网 时间:2024/05/17 16:43

问题描述:
还记得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;}
0 0