二分查找与二分答案——A-B 数对

二分查找与二分答案——A-B 数对

题目描述

出题是一件痛苦的事情！

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

好吧，题目是这样的：给出一串数以及一个数字 C，要求计算出所有 A - B = C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

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

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

输出格式

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

输入输出样例

输入样例

4 1
1 1 2 3

输出样例

3

说明/提示

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

对于 100% 的数据，1 ≤ N ≤ 2×10⁵。

保证所有输入数据都在 32 位带符号整数范围内。

分析

为了寻找满足 A - B = C 的数对的个数，我们可以先对输入的 n 个数进行排序，然后从前往后遍历，假设当前位置的值为 B，则我们只要寻找值为 B + C 的个数，最后的结果即是每个位置上所能找到的 B + C 的个数之和。而我们可以使用 lower_bound() 和 upper_bound() 这两个函数来寻找，lower_bound(begin,end,x) 用于查找第一个大于或等于 x 的数，upper_bound(begin,end,x) 用于查找第一个大于 x 的数，对于 n 个有序的数 a₁,a₂,…,a_n，设第 i 位置的值为 B，即 a_i为 B，将调用 upper_bound(a,a+n,a_i+c)-a 函数得到的结果减去调用 lower_bound(a,a+n,a_i+c)-a 函数得到的结果即为我们所要寻找的在第 i 位置上，我们可以找到的 B + C 的值的个数。

#include<bits/stdc++.h>
using namespace std;
int n,c;
int a[200050];

int main()
{
	cin>>n>>c;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	sort(a,a+n);
	long long int ans=0;
	for(int i=0;i<n;i++){
		ans+=(upper_bound(a,a+n,a[i]+c)-a)-(lower_bound(a,a+n,a[i]+c)-a);
	}
	cout<<ans<<endl;
	return 0;
}