您的位置:首页 > 大数据 > 人工智能

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

2013-09-07 10:38 393 查看
题目点此

糊里糊涂地就在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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: