騰訊2017暑假實習生招聘筆試編程題(2)

有趣的數字

---

小Q今天在上廁所時想到了這個問題：有n個數，兩兩組成二元組，差最小的有多少對呢？差最大呢？

---

輸入描述:

---

輸入包含多組測試數據。
對於每組測試數據：
N - 本組測試數據有n個數
a1,a2…an - 需要計算的數據
保證:
1<=N<=100000,0<=ai<=INT_MAX.

---

輸出描述:

---

對於每組數據，輸出兩個數，第一個數表示差最小的對數，第二個數表示差最大的對數。

---

輸入例子:

---

6
45 12 45 32 5 6

---

輸出例子:

---

1 2

---

解題思路

先對數組進行排序，然後求最大差值組和最小差值組。
最大差值肯定發生在首尾兩個數字之間，不過注意到存在相同數字，所以最大差值組數等於最小數的個數*最大數的個數
最小差值出現在相鄰兩個數之間，這裏需要考慮差值最小爲0的情況。
如果差值爲0，計算相等的數的個數n，則共有n*(n-1)/2組，不爲0的話，直接累加即可。

---

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

int main(){
    int n;
    while(cin>>n){
        vector<int> vec(n,0);
        for(int i = 0 ; i < n; i++){
            cin>>vec[i];
        }
        sort(vec.begin(),vec.end());
        int difmin = INT_MAX;
        int cntmin = 0;
        int cnt = 1;
        for(int i = 1 ;i<n;i++){
            if(vec[i]==vec[i-1]) {//差值爲0的情況
                if(difmin>0){
                    cntmin = 0;
                }
                difmin = 0;
                cnt++;  //計算個數
            }
            else {
                if(cnt>1){ // 如果存在差值爲0的情況
                    cntmin += cnt*(cnt-1)/2; 
                    cnt =1;
                }
                else {
                    int dif = vec[i]-vec[i-1];
                    if(dif<difmin){
                        difmin=dif;
                        cntmin=1;
                    }
                    else if(difmin==dif) cntmin++;                  
                }
            }

        }
        if(cnt>1){ // 當cnt大於1的時候，需要算進去
            cntmin += cnt*(cnt-1)/2;
            cnt =1;
        }
        int cnt1 = 1;
        int cnt2 = 1;
        for(int i = 1 ; i< n ;i++){
            if(vec[i]==vec[0]) cnt1++;
            else break;
        }
        for(int i = n-2; i>=0 ; i--){
            if(vec[i]==vec[n-1]) cnt2++;
            else break;
        }
        int cntmax = cnt1*cnt2;
        cout<<cntmin<<" "<<cntmax<<endl;
    }
}