给出一串正整数数列以及一个正整数 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×105,0 \leq a_i <2^{30}0≤ai<230,1 \leq C < 2^{30}1≤C<230。
2017/4/29 新添数据两组
分析,开始使用暴力枚举,部分测试数据超时,之后改用了二分查找,但是代码不简洁,就去找了一下,发现使用lower_bound和upper_bound会很简洁
#include<iostream>
#include <algorithm>
using namespace std;
const int M=1e6+5;
int a[M];
long long c;
int n,ans=0;
int main(){
scanf("%d%d",&n,&c);
for(int i=1;i<=n;i++){
scanf("%d", &a[i]);
}
sort(a+1,a+n+1);
for(int i=1;i<=n;i++){
for(int j=i+1;j<=n;j++){
if(a[j]-a[i]==c){
ans++;
}
else if(a[j]-a[i]>c)
break;
}
}
printf("%d\n",ans);
}
#include<iostream>
#include <algorithm>
using namespace std;
const int M=1e6+5;
int a[M];
long long c;
int n,ans=0;
int main(){
scanf("%d%d",&n,&c);
for(int i=1;i<=n;i++){
scanf("%d", &a[i]);
}
sort(a+1,a+n+1);
for(int i=1;i<=n;i++){
ans+=(upper_bound(a+1,a+n+1,a[i]+c)-a)-( lower_bound(a+1,a+n+1,a[i]+c)-a);
}
printf("%d\n",ans);
}
二分查找真的很好用!