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;
} 
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值