洛谷P1102 A-B数对（二分）

最新推荐文章于 2024-01-16 13:08:20 发布

Awars_zpp

最新推荐文章于 2024-01-16 13:08:20 发布

阅读量211

点赞数 1

CC 4.0 BY-SA版权

分类专栏：二分文章标签：算法

本文链接：https://blog.csdn.net/weixin_73906822/article/details/129362863

二分专栏收录该内容

6 篇文章

订阅专栏

文章介绍了如何使用二分查找算法来优化寻找满足A-B=C的正整数数对的问题。通过对给定有序数组进行二分查找，结合lower_bound和upper_bound函数（或自定义实现），可以有效地计算出数对的个数，避免了暴力求解导致的时间复杂度过高。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

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

输入格式

输入共两行。

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

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

输出格式

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

分析：

乍一看，像我一样的二分新手小蒻狗很难想到用二分查找。好的，这个题可以使用二分。

怎么二分呢，二分查找什么呢？寻找A-B=C的数对，不妨我们转化一下，寻找A=B+C的数对。不难想到枚举B，然后看一下数组中有哪些A满足。

如果暴力，复杂度为O(n^2)，似乎不能过，所以我们采用二分查找。

值得注意的是，给出的数组中是有重复数字的，也就是一个B可能对应着好几个一样的A，所以不能单纯的找到这个数，还要判断这个数的个数。其实就是我不会去重罢了。

这里我们可以使用C++提供的lower_bound（返回>= 目标值的第一个元素的地址或迭代器）和upper_bound函数（返回> 目标值的第一个元素的地址或迭代器）

#include<iostream>
#include<algorithm>
using namespace std;
long long a[200002];
long long c,ans;
int n;
int main()
{
    cin>>n>>c;
    for(int i=1;i<=n;i++) cin>>a[i];
    sort(a+1,a+n+1);
    //注意二分一定是用在有序数列
    for(int i=1;i<=n;i++)
    {
        int pos_low=lower_bound(a+1,a+n+1,a[i]+c)-a; 
        //注意“-a” 返回的是地址
        if(a[pos_low]==a[i]+c){
        //判断是否有该B对应的A
            int pos_up=upper_bound(a+1,a+n+1,a[i]+c)-a;
            ans+=pos_up-pos_low;
            //可以拿几个数试一试
        } 
    }
    cout<<ans<<endl;
    return 0;
}

注意二分一定是用在有序数列。

下面提供不使用STL的代码，其实就是把lower_bound和upper_bound写一遍。

#include<iostream>
#include<algorithm>
using namespace std;
long long c,ans;
long long a[200005];
int n;
int find_1(int l,int r,int f)
{
    while(l<=r)
    {
        int mid=l+(r-l)/2;
        if(a[mid]>=f){
            r=mid-1;
        }
        else{
            l=mid+1;
        }
    }
    if(a[l]==f) return l;
    else return -1;
}
int find_2(int l,int r,int f)
{
    while(l<=r)
    {
        int mid=l+(r-l)/2;
        if(a[mid]>f){
            r=mid-1;
        }
        else{
            l=mid+1;
        }
    }
    return l;
    //由于find2只有在find1成功找到的情况下才会使用上
    //所以不用判断 直接返回 返回一个第一个大于f的数的位置 
}
int main()
{
    cin>>n>>c;
    for(int i=1;i<=n;i++) cin>>a[i];
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)
    {
        int pos1=find_1(i+1,n,a[i]+c);
        if(pos1!=-1){
            int pos2=find_2(i+1,n,a[i]+c);
            ans+=pos2-pos1;
        }
    }
    cout<<ans<<endl;
    return 0;
}

find1寻找A的位置，find2寻找第一个大于A的位置。

看完不妨点个小赞