P1102 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。

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

#include<bits/stdc++.h>
using namespace std;
const int N=200000+10;
int n,c;
int a[N];
long long cnt;//全局变量默认值为0 

//求起始位置
int find1(int x){
	int l=1,r=n;
	while(l<r){
		int mid=l+r>>1;
		if (x<=a[mid]) r=mid;
		else l=mid+1;
	}
	if (a[l]==x) return l;
	else return -1;
}
//求结束位置 
int find2(int x){
	int l=1,r=n;
	while(l<r){
		int mid=l+r+1>>1;
		if (x>=a[mid]) l=mid;
		else r=mid-1;
	}
	if (a[l]==x) return l;
	else return -1;
}

int main(){
	cin>>n>>c;
	for (int i=1;i<=n;i++)  cin>>a[i]; //a[1]..a[n]
	sort(a+1,a+n+1);
	
	for (int i=1;i<=n;i++){ 
		int k=a[i]-c;		
		 int ans1=find1(k);//求k的起始位置
		 int ans2=find2(k);//求k的结束位置
		 if (ans1!=-1)
		   cnt=cnt+(ans2-ans1+1);
	}
	
	cout<<cnt<<endl;
	
	return 0;
}