前m大的数 (普通方法+哈希算法)

前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 <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
const int INF = 0x3fffffff;
const int MAXN = 3000+10;
const int MAXM = 5000000+10;
int ans = 0;
int arr[MAXN];
int vis[MAXM];

bool cmp(int x, int y){
    return x>y;
}
int main()
{
    int n, m;
    int len, tmp;
    while(~scanf("%d%d", &n, &m))
    {
        len = 0;
        tmp = min(n,m);
        for(int i=0; i<n; i++) scanf("%d", &arr[i]);
        sort(arr, arr+n, cmp);
        for(int i=0; i<n; i++)
            for(int j=i+1; j<n; j++){
                vis[len++] = arr[i]+arr[j];
            }
        sort(vis, vis+len, cmp);
        for(int i=0; i<m-1; i++)
            printf("%d ", vis[i]);
        printf("%d\n", vis[m-1]);
    }
    return 0;
}

哈希算法:

#include<stdio.h>
#include<stdlib.h>
#include<cstring>   //memset头文件 
#include<algorithm>
using namespace std;
const int maxn = 3005;
int a[maxn], sum[10010];
 int main()
{   int n,m;
     while (~scanf("%d %d", &n, &m))
     {
        for(int i=0; i<n; i++)
            scanf("%d", &a[i]);
        memset(sum, 0, sizeof(sum));
        for (int i=0; i<n-1; i++)
            for (int j=i+1; j<n; j++)
                 sum[a[i] + a[j]] ++;
        int count = 0;
        for (int j=10000; j>=0; j--)
        {
             if (sum[j])
             {
                 for (int i=0; i<sum[j]; i++)
                 {
                     count++;
                    count == 1 ? printf("%d", j) : printf("%d", j);
                     if (count == m)break;
                     printf(" ");
                }
             }
             if (count == m) break;
        }
        printf("\n");
     }
}

猜你喜欢

转载自blog.csdn.net/shf1730797676/article/details/81779184