腾讯2017暑期实习生编程题-有趣的数字

[编程题] 有趣的数字

时间限制：1秒

空间限制：32768K

小Q今天在上厕所时想到了这个问题：有n个数，两两组成二元组，差最小的有多少对呢？差最大呢？

输入描述:

输入包含多组测试数据。 对于每组测试数据： N - 本组测试数据有n个数 a1,a2...an - 需要计算的数据 保证: 1<=N<=100000,0<=ai<=INT_MAX.

输出描述:

对于每组数据，输出两个数，第一个数表示差最小的对数，第二个数表示差最大的对数。

输入例子1:

6
45 12 45 32 5 6

输出例子1:

1 2

首先对数组进行排序，然后统计总共有多少个不同的数字以及每个数字出现了多少次，注意特判整个数组数字相同的情况。

如果出现了相同数字，最小值就为0，即所有相同数字c(sum[i],2)之和，如果没有，遍历两遍累加即可

最大值则比较简单，为最大值个数*最小值个数

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int main(){
    int n;
    int a[101000];
    int cnt;
    int sum[101000];
    int num[101000];
    int max_num,min_num;
    while(scanf("%d",&n)!=EOF){
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
            sum[i]=1;
        }
        sort(a,a+n);
        if(a[0]==a[n-1]){
            printf("%d %d\n",n*(n-1)/2, n*(n-1)/2);
        }
        else{
            cnt = 0;
            num[cnt]=a[0];
            for(int i=1;i<n;i++){

                if(a[i]==a[i-1]){
                    sum[cnt]++;
                }
                else{
                    //cout<<a[i]<<" "<<a[i-1]<<endl;
                    num[++cnt]=a[i];
                }
            }
            cnt+=1;
            //cout<<num[cnt-1]<<endl;
            int minv =a[n-1];
            if(cnt<n){
                min_num = 0;
                for(int i=0;i<cnt;i++){
                    if(sum[i]>1){
                        min_num+=(sum[i]*(sum[i]-1)/2);
                    }
                }
            }
            else{
                min_num=0;
                for(int i=1;i<cnt;i++){
                    minv = min(minv, num[i]-num[i-1]);
                }
                for(int i=1;i<cnt;i++){
                    if(num[i]-num[i-1]==minv){
                        min_num+=1;
                    }
                }
            }
            max_num = sum[cnt-1]*sum[0];
            printf("%d %d\n",min_num, max_num);
        }
    }
    return 0;
}