A-B 数对

看题解大佬都用的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;
} 

猜你喜欢

转载自blog.csdn.net/qq_42198700/article/details/88952156
A-B