腾讯笔试题 2 有趣的数字

小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

这题需要处理的细节很多，比如，最大数里面，如何计算组合

最小的对数里面，如何计算

几个特例对了，很可能就对了

4

2 2 2 2 ->6 6

5

1 1 2 3 3->2 4

#include<iostream>
#include <stdio.h>
#include<string>
#include <vector>
#include<algorithm>
using namespace std;

#define INT_MAX 2147483647

int main()
{
	
	long N;
	while(cin>>N) 
	{
		vector<long> a(N,0);
		for(long i = 0; i < N;i++)
			cin>>a[i];
		sort(a.begin(),a.end());
		long max_cha;
		long cha_head = 1;
		long cha_tail = 1;
		for(long i =0;i < N;i ++)
		{
			if(i!= 0)
			{
				if(a[i] == a[i-1])
					cha_head++;
				else
					break;
			}
			
		}
		for(long i = 0;i < N;i++)
		{
			if(i!=0)
			{
				if(a[N-i-1] == a[N-i])
					cha_tail ++;
				else
					break;
			}
		}
		if(cha_tail == N)//全部数据一致
			max_cha = N*(N-1)/2;
		else
			max_cha = cha_head*cha_tail;
		//最小
		long min_cha = 0;
		long min = INT_MAX;
		long lianxu = 1;
		for(long i =0;i < N;i ++)
		{
			if(i!= 0)
			{
				if((a[i] - a[i-1]) == 0)
					lianxu ++;
				else if(lianxu > 1)
				{
					if(min == 0)
						min_cha+= lianxu * (lianxu - 1)/2;
					else
					{
						min_cha = lianxu * (lianxu - 1)/2;
						min = 0;
					}
					lianxu = 1;
				}
				else if((a[i] - a[i-1]) == min)
					min_cha ++;
				else if((a[i] - a[i-1]) < min)
				{
					min_cha = 1;
					min = (a[i] - a[i-1]);
				}
			}
		}
		if(lianxu > 1)
		{
			if(min == 0)
				min_cha+= lianxu * (lianxu - 1)/2;
			else
			{
				min_cha = lianxu * (lianxu - 1)/2;
				min = 0;
			}
		}
		cout<<min_cha<<" "<<max_cha<<endl;
	}

	return 0;

}