温故而知新

数对

题目描述

出题是一件痛苦的事情！

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

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

输入共两行。

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

第二行，NNN 个整数，作为要求处理的那串数。
输出格式

一行，表示该串数中包含的满足 A−B=CA - B = CA−B=C 的数对的个数。
输入输出样例
输入 #1

4 1
1 1 2 3

输出 #1

3

说明/提示

对于 75%75%75% 的数据，1≤N≤20001 \leq N \leq 20001≤N≤2000。

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

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

2017/4/29 新添数据两组

思路

先排序，再用二分upper_bound()-lower_bound()就是每次满足条件所有的相同的‘B’了。

代码

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n,c;
	cin>>n>>c;
	long long a[200005];
	for(int i=1;i<=n;i++) 
	{cin>>a[i];
	}
	sort(a+1,a+n+1);
	long ans=0;
	for(int i=1;i<=n;i++)
	{
		ans+=(upper_bound(a+1,a+1+n,a[i]-c)-lower_bound(a+1,a+1+n,a[i]-c));
	}
	cout<<ans;
}