洛谷——P1102 A-B 数对

题目背景

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！

题目描述

给出一串正整数数列以及一个正整数 C C C，要求计算出所有满足 A − B = C A - B = C A−B=C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个正整数 N , C N,C N,C。

第二行， N N N 个正整数，作为要求处理的那串数。

输出格式

一行，表示该串正整数中包含的满足 A − B = C A - B = C A−B=C 的数对的个数。

样例 #1

样例输入 #1

4 1
1 1 2 3

样例输出 #1

3

提示

对于 75 % 75\% 75% 的数据， 1 ≤ N ≤ 2000 1 \leq N \leq 2000 1≤N≤2000。

对于 100 % 100\% 100% 的数据， 1 ≤ N ≤ 2 × 1 0 5 1 \leq N \leq 2 \times 10^5 1≤N≤2×105， 0 ≤ a i < 2 30 0 \leq a_i <2^{30} 0≤ai​<230， 1 ≤ C < 2 30 1 \leq C < 2^{30} 1≤C<230。

代码

#include<bits/stdc++.h>
using namespace std;
long long n, c, ans, a[200005];
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin >> n >> c;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	sort(a + 1, a + n + 1);
	for (int i = 1; i <= n; i++) {
		ans += ((upper_bound(a + 1, a + n + 1, a[i] + c) - a) - (lower_bound(a + 1, a + n + 1, a[i] + c) - a));
	}
	cout << ans;
	return 0;
}

AC合格证