P1102 A-B 数对

题目背景

出题是一件痛苦的事情！

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

题目描述

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

输入格式

输入共两行。

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

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

输出格式

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

样例 #1

样例输入 #1

4 1
1 1 2 3

样例输出 #1

3

提示

对于 $75\%$ 的数据，$1\le N\le 2000$。

对于 $100\%$ 的数据，$1\le N\le 2\times 10^5$，$0\le a_i<2^{30}$，$1\le C<2^{30}$。

思路

双指针算法，具体见代码

代码

#include<iostream>
#include<algorithm>
#include<cstring>
#define endl '\n'
using namespace std;

typedef long long LL;// 会爆int！！！见解析1

const int N = 200010;

int n, c;
int a[N];

int main() {
	ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
	cin >> n >> c;
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}

	sort(a, a + n);// 排序，使序列单调递增

	LL res = 0;

	// 双指针算法
	for (int i = 1, j = 0; i < n; i++) {

		// 从前向后找第一个 可能 满足a[i] - a[j] = c的j
		while (j < i && a[i] - a[j] > c) {
			j++;
		}

		LL res0;

		if (j < i && a[i] - a[j] == c) {

			// 由于序列中存在重复的数字，不同位置的数字一样的数对算不同的数对
			// 即满足条件的 i 和 j 可能有多对
			// res0存储的就是当找到第一个满足条件的 i 时，对应的 j 有多少个
			res0 = 0;
			while (j < i && a[i] - a[j] == c) {
				res0++;
				res++;
				j++;
			}

			// 看同样满足条件的 i 有多少个
			i++;
			while (i < n && a[i] == a[i - 1]) {
				res += res0;
				i++;
			}
			i--;// 最外层的for循环i会++
		}
	}
	cout << res << endl;
	return 0;
}

解析

1. 这题还是没有注意到会爆int。。。

样例输入 #2

200000 1
1 1 1 ... 1 1 1 2 2 2 ... 2 2 2

即前100000个是1，后100000个是2，那么此时A-B数对的个数会达到$10^5\ast 10^5=10^{10}$，会爆int！因此要用long long。

2. 此题的其他解法：二分、哈希