二分查找啊

最新推荐文章于 2024-02-20 14:28:52 发布

q619718

最新推荐文章于 2024-02-20 14:28:52 发布

阅读量540

点赞数

文章标签：算法开发语言

本文链接：https://blog.csdn.net/q619718/article/details/127408433

版权

这篇博客讨论了一种解决数对计数问题的方法，通过将数组排序然后利用二分查找来提高效率。原始的暴力枚举方法在处理大数据时可能导致超时，而二分查找的引入显著提升了算法的性能。博主分享了两种实现，一种是双重循环，另一种是利用`lower_bound`和`upper_bound`函数，后者代码更简洁且运行更快。

摘要由CSDN通过智能技术生成

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

输入格式

输入共两行。

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

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

输出格式

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

输入输出样例

输入 #1复制
4 1
1 1 2 3
输出 #1复制
3
说明/提示

对于 75\%75% 的数据，1 \leq N \leq 20001≤N≤2000。

对于 100\%100% 的数据，1 \leq N \leq 2 \times 10^51≤N≤2×105，0 \leq a_i <2^{30}0≤ai<230,1 \leq C < 2^{30}1≤C<230。

2017/4/29 新添数据两组

分析,开始使用暴力枚举,部分测试数据超时,之后改用了二分查找,但是代码不简洁,就去找了一下,发现使用lower_bound和upper_bound会很简洁

#include<iostream>
#include <algorithm>
using namespace std;
const int M=1e6+5;
int a[M];
long long c;
int n,ans=0;
int main(){
    scanf("%d%d",&n,&c);
    for(int i=1;i<=n;i++){
        scanf("%d", &a[i]);
    }
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++){
        for(int j=i+1;j<=n;j++){
            if(a[j]-a[i]==c){
                ans++;
            }
            else if(a[j]-a[i]>c)
                break;
        }
    }
    printf("%d\n",ans);

}

#include<iostream>
#include <algorithm>
using namespace std;
const int M=1e6+5;
int a[M];
long long c;
int n,ans=0;
int main(){
    scanf("%d%d",&n,&c);
    for(int i=1;i<=n;i++){
        scanf("%d", &a[i]);
    }
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++){
        ans+=(upper_bound(a+1,a+n+1,a[i]+c)-a)-( lower_bound(a+1,a+n+1,a[i]+c)-a);
    }
    printf("%d\n",ans);

}

二分查找真的很好用!