p1102 A-B数对

A-B 数对

题目背景

出题是一件痛苦的事情！

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

题目描述

给出一串正整数数列以及一个正整数 \(C\)，要求计算出所有满足 \(A - B = C\) 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个正整数 \(N,C\)。

第二行，\(N\) 个正整数，作为要求处理的那串数。

输出格式

一行，表示该串正整数中包含的满足 \(A - B = C\) 的数对的个数。

样例 #1

样例输入 #1

4 1
1 1 2 3

样例输出 #1

3

提示

对于 \(75\%\) 的数据，\(1 \leq N \leq 2000\)。

对于 \(100\%\) 的数据，\(1 \leq N \leq 2 \times 10^5\)，\(0 \leq a_i <2^{30}\),\(1 \leq C < 2^{30}\)。

2017/4/29 新添数据两组

思路

考虑到A-B=C情况可以转化为A-C=B，此时先将输入的情况排序，枚举每一个A的情况，如果存在第一个符合这一条件的数，定为左边；再找最后一个符合这一条件的数定为右边，ans加上r-l即可。

完整代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<set>
#include<unordered_map>
using namespace std;
int n,c;
int a[200005]; 
int main(){
	long long ans=0;
	cin>>n>>c;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	
	sort(a+1,a+n+1);
	int r1=1;
	int r2=1;
	for(int i=1;i<=n;i++){
		while(r1<=n&&a[r1]-a[i]<c)
			++r1;
		while(r2<=n&&a[r2]-a[i]<=c)
			++r2;
		if(a[r1]-a[i]==c&&a[r2-1]-a[i]==c)
			ans+=r2-r1;
	}
	cout<<ans;
	return 0;
}