带注释 P1102 A-B 数对

题目描述
出题是一件痛苦的事情！

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

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

输入格式
输入共两行。

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

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

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

输入输出样例
输入 #1

4 1
1 1 2 3

输出 #1

3

说明/提示
对于 75%75% 的数据，1 \leq N \leq 20001≤N≤2000。

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

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

2017/4/29 新添数据两组

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[200010],c;
//二分函数，找出第一个符合条件的元素 
int bsearch1(int as,int l,int r){
	while (l < r){
		int mid=l+r >> 1;
		if (a[mid] >= as-c) r=mid;
		else l=mid+1;
	}
	return l;
}
//找出最后一个 
int bsearch2(int as,int l,int r){
	while (l < r){
		int mid=l+r+1 >> 1;
//因为l=mid记得k+r+1 
		if (a[mid] <= as-c) l=mid;
		else r=mid-1;
	}
	return l;
}
int main (){
	int n;
	scanf ("%d %d",&n,&c);
	int i,j,k;
	long int cnt=0;
	for (i = 0;i < n;i++) scanf ("%d",&a[i]);
	sort (a,a+n);
	for (i = n-1;i >= 0;i--) {
		int c1=bsearch1(a[i],0,n-1);
		int c2=bsearch2(a[i],0,n-1);
		if (a[i] - a[c1] == c) cnt+=c2-c1+1; 
//符合条件，计上 
	}
	printf ("%ld",cnt);
//由于测试数据加强， cnt要long int 
	return 0;
}