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; }
补充:软件开发 , C++ ,