luogu P1102 A-B 数对 |二分查找

题目描述

出题是一件痛苦的事情！

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

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

输入格式

第一行包括22个非负整数N和C，中间用空格隔开。

第二行有N个整数，中间用空格隔开，作为要求处理的那串数。

输出格式

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

说明/提示

对于73%73%的数据，N≤2000；

对于100%100%的数据，N≤200000。

所有输入数据都在longint范围内。

---

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define int long long
using namespace std;
const int N=2e5+10;
int A[N];
signed main(){
	int n,c;
	cin>>n>>c;
	for(int i=1;i<=n;i++)
	scanf("%lld",A+i);
	sort(A+1,A+n+1);
	int ans=0;
	for(int i=1;i<=n;i++){
		int a=c+A[i];
		int op=upper_bound(A+1,A+1+n,a)-A;
		op-=lower_bound(A+1,A+1+n,a)-A;
		ans+=op;
	}
	cout<<ans<<endl;
}