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

来源:互联网 发布:淘宝如何修改发货地址 编辑:程序博客网 时间:2024/05/29 16:31

题目点此

糊里糊涂地就在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;}


原创粉丝点击