5 .A-B 数对-二分查找/模拟（普及-）

问题描述

出题是一件痛苦的事情！
相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！
好吧，题目是这样的：给出一串数以及一个数字 C，要求计算出所有A-B=C的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式
输入共两行。
第一行，两个整数 N,C。
第二行，N个整数，作为要求处理的那串数。

输出格式
一行，表示该串数中包含的满足A-B=C的数对的个数。

输入输出样例

输入 #1

4 1
1 1 2 3

输出 #1

3

说明/提示
对于75%的数据，1≤N≤2000。
对于100%的数据，1≤N≤2×10^5
保证所有输入数据绝对值小于 2^30，且C≥1。

问题分析

  这道题主要考察查找和模拟，我们可以通过先将所有数字从小到大排序，再查找A数的后面所有数字，找到满足A-B=C的B数，B数的个数即是该A数的所有满足条件的数对个数，A数通过遍历所有数字获得即可。这里我刚开始查找每一个B数的方法利用的是二分法，获取B数的个数使用的是简单遍历相加获取，最后发现程序超时。便改进查找和获取B数个数的方法，可以利用双指针的方法，一个指针指向所有B数的开端，一个指针指向所有B数的结尾，B数的个数即是两个指针的相减数，查找方法则可以直接使用遍历查找，程序成功AC。

代码实现

#include <iostream>

#include <algorithm>

using namespace std;

int main()
{
	int r1 = 1, r2 = 1;								//双指针
	int n, c;
	long long ans = 0;
	cin >> n >> c;
	int* a = new int[n];
	for (int i = 0; n > i; i++)
		cin >> a[i];
	sort(a, a + n);									//从小到大排序
	for (int i = 0; n > i; i++)
	{
		while (a[r1] - a[i] < c && r1 < n)			//找到B数的开端
			r1++;
		while (a[r2] - a[i] <= c && r2 < n)			//找到B数的结尾
			r2++;
		ans += r2 - r1;								//该A数满足条件的所有数对个数
	}
	cout << ans;
	delete[]a;
	return 0;
}

运行结果

总结

  这道题其实还可以用哈希表的方法来查找B数的个数，也比较方便，这道题总的来说主要还是考察查找和模拟，比较简单。