HackerRank Algorithms-Search-Pairs 数组的固定差对数

2014-11-23 22:08:34 来源: 作者: 浏览: 2
糊里糊涂地就在HackerRank做了几题水题,姑且发出来吧。。。
题意:给出n和k,然后给出n个数,求出数组里面两两数的差为k有几对。
其中[N<=10^5], [K>0 and K<1e9],所有数都在int范围内,很明显,如果裸暴力的话肯定会超时。(虽然HackerRank好像没有时间限制,越快分数越高。。。)
于是换种写法,遍历数组然后查询该数+k有没有存在,存在就累计+1,储存和查询用set实现,复杂度为log(n),总的复杂度是nlog(n),应该会快很多。
代码:
 
 /* 
 *   Author:        illuz  
 *   Blog:          http://blog.csdn.net/hcbbt 
 *   File:          pairs.cpp 
 *   Lauguage:      C/C++ 
 *   Create Date:   2013-09-06 09:06:47 
 *   Descripton:    pairs, sort  
 */  
#include   
#include   
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 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::iterator i = m.begin(); i != m.end(); i++) {  
        t = *i;  
        if (m.count(t + k))  
           cnt++;     
    }  
    printf("%d\n", cnt);  
    return 0;  
}  


-->

评论

帐  号: 密码: (新用户注册)
验 证 码:
表  情:
内  容: