前m大的数 (经典的思维题)

题目链接:https://vjudge.net/contest/247842#problem/A

前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

先把所有的数字相加,通过下标避免了sort排序;

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

const int _INF = -1e9-1;

int vec[10005];
int num[3005];

int main(){
    int n,m;
    while(cin>>n>>m){
        memset(vec,0,sizeof(vec));
        for(int i=0;i<n;i++){
            scanf("%d",&num[i]);
        }
        int maxn=_INF;
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                if(num[i]+num[j]>maxn){
                    maxn=num[i]+num[j];
                }
                vec[num[i]+num[j]]++;
            }
        };
        if(m<=0){
            continue;
        }
        printf("%d",maxn);
        vec[maxn]--;
        m--;
        for(int i=maxn;i>=0;i--){
            if(m==0){
                break;
            }
            if(vec[i]>0){
                    while(vec[i] && m>0){
                        printf(" %d",i);
                        vec[i]--;
                        m--;
                    }
            }
        }
        cout<<endl;

    }




    return 0;
}

猜你喜欢

转载自blog.csdn.net/guozuofeng/article/details/81778697