A-B数对（二分）

1. 处理mid=(l+r)/2防越界问题

改进：mid=l+(r-l)/2

2.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
2
输出 #1

3
1
说明/提示
 

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

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

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

#include<iostream>
#include<algorithm> 
using namespace std;
int n,c;
int a[200010];
int binarySearch1(int a1[],int n1,int b)
{
	int l=1,r=n1;
	int ans=0;
	while(l<=r)
	{
		int m=(l+r)/2;
		if(a1[m]<=b)
		{
			l=m+1;
		} 
		else
		{
			ans=m;
			r=m-1;
		}
	}
	return ans;
}
int binarySearch2(int a1[],int n1,int b)
{
	int l=1,r=n1;
	int ans=0;
	while(l<=r)
	{
		int m=(l+r)/2;
		if(a1[m]>=b)
		{
			ans=m;
			r=m-1;
		} 
		else
		{
			l=m+1;
		}
	}
	return ans;
}
int main()
{
	
	cin>>n>>c;
	int ans=0;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	sort(a+1,a+1+n);
	for(int i=1;i<=n;i++)
	{
		int b=a[i]-c;
		//二分查找，在数组a中查找b的个数
		ans+=binarySearch1(a,n,b)-binarySearch2(a,n,b); 
	}
	cout<<ans;
	
	return 0;
}

还有一种更简单的方法，用map记录一下每个数出现的次数，上代码：

#include<iostream>
#include<algorithm> 
#include<map>
using namespace std;
typedef long long ll;
int a[200010];
map<int,int>s;
ll ans;
int main()
{
	int n,c;
	cin>>n>>c;
	for(int i=1;i<=n;i++)
	{
	
		cin>>a[i];
		s[a[i]]++;
	}
	sort(a+1,a+1+n);
	for(int i=n;i>=1;i--)
	{
		if(s[a[i]]&&s[a[i]-c])
		{
			ans+=(ll)s[a[i]]*s[a[i]-c];
			s[a[i]]=0;
		}
	}
	cout<<ans;
	
	return 0;
}