HackerRank Algorithms-Search-Pairs 數組的固定差對數

糊裡糊塗地就在HackerRank做了幾題水題，姑且發出來吧。。。 題意：給出n和k，然後給出n個數，求出數組裡面兩兩數的差為k有幾對。 其中[N<=10^5], [K>0 and K<1e9]，所有數都在int范圍內，很明顯，如果裸暴力的話肯定會超時。（雖然HackerRank好像沒有時間限制，越快分數越高。。。） 於是換種寫法，遍歷數組然後查詢該數+k有沒有存在，存在就累計+1，儲存和查詢用set實現，復雜度為log(n)，總的復雜度是nlog(n)，應該會快很多。 代碼：

 
 /* 
 *   Author:        illuz <iilluzen[at]gmail.com> 
 *   Blog:          http://blog.csdn.net/hcbbt 
 *   File:          pairs.cpp 
 *   Lauguage:      C/C++ 
 *   Create Date:   2013-09-06 09:06:47 
 *   Descripton:    pairs, sort  
 */  
#include <cstdio>  
#include <set>  
using namespace std;  
#define rep(i, n) for (int i = 0; i < (n); i++)  
#define FI(i, x) for (typeof((x).begin()) i = (x).begin(); i != (x).end(); i++)  
  
set<int> m;  
  
int main() {  
    int n, k, t;  
    scanf("%d%d", &n, &k);  
    rep(i, n) {  
        scanf("%d", &t);  
        m.insert(t);  
    }  
    int cnt = 0;  
    for (set<int>::iterator i = m.begin(); i != m.end(); i++) {  
        t = *i;  
        if (m.count(t + k))  
           cnt++;     
    }  
    printf("%d\n", cnt);  
    return 0;  
}