P1102 A-B 数对

P1102 A-B 数对

题目背景

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！

题目描述

给出一串正整数数列以及一个正整数 $C$，要求计算出所有满足 $A-B=C$ 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个正整数 $N,C$。

第二行，$N$ 个正整数，作为要求处理的那串数。

输出格式

一行，表示该串正整数中包含的满足 $A-B=C$ 的数对的个数。

样例

样例输入

4 1
1 1 2 3

样例输出

3

提示

对于 $75\%$ 的数据，$1\le N\le 2000$。

对于 $100\%$ 的数据，$1\le N\le 2\times 10^5$，$0\le a_i<2^{30}$，$1\le C<2^{30}$。

暴力枚举的code:

#include<iostream>
using namespace std;
const int N=2*10e5+10;
int n,c,cnt;
int arr[N];
int main(){
    cin>>n>>c;
    for(int i=1;i<=n;i++){
        cin>>arr[i];
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(arr[i]-arr[j]==c){
                cnt++;
            }
        }
    }
    cout<<cnt;
    return 0;
}

二分的code：

把A-B=C看成A=B+C，然后枚举B，查找A。

#include<iostream>
#include<algorithm>
using namespace std;
const int N=2*10e5+10;
int n,c,res1,res2;
long long cnt;
int arr[N];
int binary_search1(int x){
    int l=0,r=n+1;
    while(l+1<r){
        int mid=(l+r)/2;
        if(arr[mid]<x){
            l=mid;
        }
        else r=mid;
    }
    if(arr[r]==x) return r;
    else return -1;
}
int binary_search2(int x){
    int l=0,r=n+1;
    while(l+1<r){
        int mid=(l+r)/2;
        if(arr[mid]<=x){
            l=mid;
        }
        else r=mid;
    }
    if(arr[l]==x) return l;
    else return -1;
}
int main(){
    cin>>n>>c;
    for(int i=1;i<=n;i++){
        cin>>arr[i];
    }
    sort(arr+1,arr+1+n);
    for(int b=1;b<=n;b++){
        int a=arr[b]+c;
        res1=binary_search1(a);
        if(res1==-1) continue;
        else{
            res2=binary_search2(a);
            cnt+=res2-res1+1;
        }
    }
    cout<<cnt<<endl;
    return 0;
}

二分容易错误的代码：

#include<iostream>
using namespace std;
const int N=200010;
int n,c;
int arr[N];
int res=0;
int main(){
    cin>>n>>c;
    for(int i=1;i<=n;i++){
        cin>>arr[i];
    }
    for(int i=1;i<=n;i++){
        int A=arr[i]+c;
        int l=0,r=n+1;
        while(l+1<r){
            int mid=(l+r)/2;
            if(arr[i]<=A) l=mid;
            else r=mid;
            if(arr[l]==A) res++;
        }
    }
    cout<<res;
}