题源 C已知,C+B=A,枚举所有的B,在数组中找A的个数,巧妙之处就是得用STL的upper_bound和lower_bound。不过也能用双指针
题目背景
出题是一件痛苦的事情!
相同的题目看多了也会有审美疲劳,于是我舍弃了大家所熟悉的 A+B Problem,改用 A-B 了哈哈!
题目描述
给出一串正整数数列以及一个正整数 C,要求计算出所有满足 A−B=C 的数对的个数(不同位置的数字一样的数对算不同的数对)。
输入格式
输入共两行。
第一行,两个正整数 N,C。
第二行,N 个正整数,作为要求处理的那串数。
输出格式
一行,表示该串正整数中包含的满足 A−B=C 的数对的个数。
输入输出样例
输入 #1复制
4 1 1 1 2 3输出 #1复制
3说明/提示
对于 75% 的数据,1≤N≤2000。
对于 100% 的数据,1≤N≤2×10^5,0≤ai<2^30,1≤C<2^30。
upper_bound和lower_boun方法
#include<bits/stdc++.h>
using namespace std;
int n,c;
int a[1000009]={0};
long long ant=0;
int main()
{
cin>>n>>c;
for(int i=1;i<=n;i++){
cin>>a[i];
}
sort(a+1,a+1+n);
for(int i=1;i<=n;i++){
ant+=upper_bound(a+1,a+1+n,a[i]+c)-lower_bound(a+1,a+1+n,a[i]+c);
}
cout<<ant;
return 0;
}
双指针
#include<bits/stdc++.h>
using namespace std;
int n,c;
int a[1000009]={0};
long long ant=0;
int main()
{
cin>>n>>c;
for(int i=1;i<=n;i++){
cin>>a[i];
}
sort(a+1,a+1+n);
int L=1,L1=1;
for(int i=1;i<=n;i++){
while(L<=n&&a[L]-a[i]<c){
L++;
}
while(L1<=n&&a[L1]-a[i]<=c){
L1++;
}
if(a[i]+c==a[L1-1]&&a[i]+c==a[L]){
ant+=L1-L;
}
// cout<<L<<" "<<L1<<" "<<ant<<endl;
}
cout<<ant;
return 0;
}