数对
题目描述
出题是一件痛苦的事情!
相同的题目看多了也会有审美疲劳,于是我舍弃了大家所熟悉的 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;
}