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

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

输入描述:

输入包含多组测试数据。

对于每组测试数据：

N - 本组测试数据有n个数

a1,a2…an - 需要计算的数据

保证:

1<=N<=100000,0<=ai<=INT_MAX.

输出描述:

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

示例1

输入

6

45 12 45 32 5 6

输出

1 2

思路：分两个问题求解，首先都是将原数组排序。

然后求解差最大的对数，算出最小的数的个数和最大的数的个数，相乘以后的结果即是差最大的对数。

最后求解差最小的对数，分两种情况讨论，差最小是否为0.

如果差最小不为0，计算相邻数之间的差，统计差最小的对数。

如果差最小为0，计算原数组中重复数的个数，每一种数字重复即算一次组合数n*(n-1)/2，最后累加就是差最小的对数。

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
void solution(int n,vector<int> nums){
int min=0,max=0;
sort(nums.begin(),nums.end());
int begin = 0;
int end = 0;
for(int i = 0;i<n;i++){
if(nums[i] == nums[0])
begin++;
else break;
}
for(int i = n-1;i>=0;i--){
if(nums[i] == nums[n-1])
end++;
else break;
}
max = begin * end;
vector<int> temp;
for(int i=0;i<n-1;i++){
temp.push_back(nums[i+1]-nums[i]);
}
sort(temp.begin(),temp.end());
if(temp[0]){
for(int i=0;i<temp.size();i++){
if(temp[0]==temp[i]) min++;
else break;
}
}
else{
int count = 1;
for(int i = 1;i<n;++i){
if(nums[i] == nums[i-1])
count++;
else{
min += count*(count-1)/2;
count = 1;
}
}
min += count*(count-1)/2;
}
cout<<min<<' '<<max<<endl;
}
int main(){
int n;
while(cin>>n)
{
int temp=0;
vector<int>nums;
for (int i=0;i<n;i++)
{
cin>>temp;
nums.push_back(temp);
}
solution(n,nums);
}
return 0;
}