P1102 A-B 数对

最新推荐文章于 2023-03-24 17:24:52 发布

原创最新推荐文章于 2023-03-24 17:24:52 发布 · 560 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

洛谷官方题单专栏收录该内容

6 篇文章

订阅专栏

这是一道算法题目，要求根据输入的数列和目标差值C，计算所有满足A-B=C的数对数量。输入包括两行，第一行包含整数N和C，第二行包含N个整数。例如，对于输入4 1和1 1 2 3，输出为3，因为有三个数对(1,2), (2,1), (3,2)满足A-B=1。" 45815445,5025379,精仿京东商城Android客户端源码分享,"['Android开发', '移动开发', '电商应用', '源码分享']

[题目描述]

出题是一件痛苦的事情！
相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！
好吧，题目是这样的：给出一串数以及一个数字 C，要求计算出所有 A−B=C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

[输入格式]

输入共两行。
第一行，两个整数 N,C。
第二行，N 个整数，作为要求处理的那串数。

[输出格式]

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

[输入样例]

输入 #1
4 1
1 1 2 3

[输出样例]

输出 #1
3

【源程序】

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=2e5+10;
int n,cnt;
LL res;
int f[N];

int main(){
    cin>>n>>cnt;
    int l,r,mid;
    for(int i=1;i<=n;i++)cin>>f[i];
    
    sort(f+1,f+1+n);

    for(int i=1;i<=n;i++){
        int ver=f[i]-cnt;
        l=1,r=i-1;
        while(l<r){
            mid=l+r>>1;
            if(f[mid]>=ver)r=mid;
            else l=mid+1;
        }
        if(f[l] != ver)continue;
        else{
            int fir=l;
            l=fir,r=i-1;
            while(l<r){
                mid=l+r+1>>1;
                if(f[mid]<=ver)l=mid;
                else r=mid-1;
            }
            res+=l-fir+1;
        }
    }
    
    cout<<res<<endl;
    return 0;
}