编程题—有趣的数字

内容会持续更新，有错误的地方欢迎指正，谢谢!

题目

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

输入描述:
输入包含多组测试数据。

对于每组测试数据：
N - 本组测试数据有n个数
a1,a2…an - 需要计算的数据

保证:
1<=N<=100000,0<=ai<=INT_MAX.

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

输入例子:
6
45 12 45 32 5 6

输出例子:
1 2

分析

开始感觉挺简单，直接O（n^2）就完了，但求最大值的对数的时候，O(n^2)超时了。。。

思路：

1. 先排序
2. 求差为最小值的对数
   
   1. 先求最小值minVal
   2. 若存在重复的元素：差为0，双重循环求差为0的对数，即得到minCount
   3. 若不存在重复的元素：最小值只出现在相邻元素间，循环一次即可得到minCount
3. 求差为最大值的对数
   
   1. 若全部元素都相等，利用排列组合的知识：maxCount=x*(x-1)/2
   2. 若不是全部元素都相等：maxCount=最小元素的个数*最大元素的个数

注意：千万不要忽略重复元素的情况！比如：3 3 3 3，最大值对是6对！而不是3对，更不是12对！

所用数据结构： vector

代码

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int main()
{
    int num,i,j;
    while(cin>>num)//读入元素的总个数
    {
        if(num<2)
        {
            cout<<0<<" "<<0<<endl;
            continue;
        }
        vector<int> arr;//arr不要声明为全局变量,不然全部测试数据都被存入了arr
        int length=num;
        int temp;
        while(num--)//不能写成while(cin>>temp)，不然，全部测试数据都被一次性存入了arr
        {
            cin>>temp;
            arr.push_back(temp);
        }
        sort(arr.begin(),arr.end());//C++排序函数：对[begin,end)内所有元素排序

        //求最小值minVal
        int minVal=arr[1]-arr[0];
        for(i=1;i<length-1;++i)
        {
            int cur=arr[i+1]-arr[i];
            if(cur<minVal)
                minVal=cur;
        }
        //求最小值的个数minCount
        int minCount=0;
        if(minVal==0)//arr中有相等元素时
        {
            for(i=0;i<length-1;++i)
            {
                for(j=i+1;j<length;++j)
                {
                    if(arr[i]==arr[j])
                        ++minCount;
                }
            }
        }
        else//arr中无元素相等时
        {
            for(i=0;i<length-1;++i)
            {
                int cur=arr[i+1]-arr[i];
                if(cur==minVal)
                {
                    ++minCount;
                }
            }
        }

        //求最大值maxVal
        int maxVal=arr[length-1]-arr[0];
        //求最大值的个数maxCount
        int maxCount=0;
        if(maxVal==0)//全部元素都相等,利用组合原理
        {
            maxCount=length*(length-1)/2;
        }
        else//有不同的元素,最大值的个数=最小的个数*最大的个数
        {
            int smallCount=0,bigCount=0;
            for(i=0;i<length;++i)
            {
                if(arr[i]==arr[0])
                    ++smallCount;
                else if(arr[i]==arr[length-1])
                    ++bigCount;
            }
            maxCount=smallCount*bigCount;
        }
        cout<<minCount<<" "<<maxCount<<endl;
    }
    return 0;
}