看题解大佬都用的map映射,蒟蒻不会map,排序来顶
NlogN的做法
对于本题,开始直接无脑暴力,居然还84分,惊!
后来发现优化也很容易,排序加二分,最普通的做法了吧。 开个离散化后的B数组,统计每个数的数量,然后二分查找,如果满足A-B==C,加上此位置的数量,详见代码段
开离散化的B数组,在C++的STL中有unique函数
for(register int i=1;i<=n;i++)
scanf("%lld",&a[i]),b[i]=a[i];
sort(b+1,b+n+1);
q=unique(b+1,b+n+1)-b-1;//离散化
sort(a+1,a+n+1);//排序A数组,以便于后面统计数量
此时a,b都是有序的,开个下标 O(n)统计数量
int fl=1;//下标
for(int i=1;i<=n;i++){
if(a[i]==b[fl]){
f[fl]++;
} else{
fl++;//下标移动
if(a[i]==b[fl]) f[fl]++;//移动后再次进行比较
}
}
nlogn查找就行了
for(register int i=1;i<=n;i++){
int l=1,r=q;
while(l<=r){//普通二分
int mid=(l+r)/2;
if(a[i]-b[mid]<c) r=mid-1;
else l=mid+1;
}
if(a[i]-b[r]==c){
ans+=f[r];//统计数量
}
}
全部代码
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=200007;
long long n,c,a[N],ans,b[N],f[N];
int l,r,q;
int main(){
scanf("%lld%lld",&n,&c);
for(register int i=1;i<=n;i++)
scanf("%lld",&a[i]),b[i]=a[i];
sort(b+1,b+n+1);
q=unique(b+1,b+n+1)-b-1;
sort(a+1,a+n+1);
int fl=1;
for(int i=1;i<=n;i++){
if(a[i]==b[fl]){
f[fl]++;
} else{
fl++;
if(a[i]==b[fl]) f[fl]++;
}
}
for(register int i=1;i<=n;i++){
int l=1,r=q;
while(l<=r){
int mid=(l+r)/2;
if(a[i]-b[mid]<c) r=mid-1;
else l=mid+1;
}
if(a[i]-b[r]==c){
ans+=f[r];
}
}
printf("%lld",ans);
return 0;
}