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×，0≤<，1≤C<。

思路

要找a-b=c的数对个数，仅需要找到a=b+c数对个数，可以用二分查找，其实就是对于每个a[i]找到所有a[i]+c的起始位置和终止位置做差然后求和就是答案。

#include<iostream>
#include<algorithm>
using namespace std;
const int N=200010;
long long a[N];
long long n,x;
int find1(int x){//查找大于等于a[i]+c的最小下标 
	int l=1,r=n;
	while(l<r){
		int mid=(l+r+1)/2;
		if(a[mid]<=x)l=mid;
		else r=mid-1;
	}
	return l;
}
int find2(int x){//查找小于a[i]+c的最大下标 
	int l=1,r=n;
	while(l<r){
		int mid=(l+r+1)/2;
		if(a[mid]<x)l=mid;
		else r=mid-1;
	}
	return l;
} 
int main(){
	cin>>n>>x;
	for(int i=1;i<=n;i++)cin>>a[i];
	sort(a+1,a+n+1);
	long long ans=0;
	for(int i=1;i<=n;i++){
		ans+=find1(a[i]+x)-find2(a[i]+x);
	}
	cout<<ans;
	return 0; 
}

当然也可以用lower_bound/upper_bound来做，这两个函数是stl中用于二分查找的函数。

假定我们有一个有序的数组a，并将数x作为二分查找的目标

lower_bound(a,a+n,x)-a      //下标从0开始
lower_bound(a+1,a+n+1,x)-a  //下标从1开始

lower_bound的作用是返回大于等于x的最小的下标

upper_bound(a,a+n,x)-a      //下标从0开始
upper_bound(a+1,a+n+1,x)-a  //下标从1开始

upper_bound的作用是返回大于x的最小下标 

参考代码：

#include<iostream>
#include<algorithm>
using namespace std;
const int N=200010;
long long a[N];
long long n,x;
int main(){
	cin>>n>>x;
	for(int i=1;i<=n;i++)cin>>a[i];
	sort(a+1,a+n+1);
	long long ans=0;
	for(int i=1;i<=n;i++){
		ans+=(upper_bound(a+1,a+n+1,a[i]+x)-a)-(lower_bound(a+1,a+n+1,a[i]+x)-a);
	}
	cout<<ans;
	return 0; 
}