图解二分法（二分查找）（Aswing 789. 数的范围）

最新推荐文章于 2024-02-21 18:22:28 发布

十二月的猫

最新推荐文章于 2024-02-21 18:22:28 发布

阅读量705

点赞数 26

分类专栏：算法文章标签：算法数据结构

本文链接：https://blog.csdn.net/m0_67656158/article/details/134397765

版权

算法专栏收录该内容

22 篇文章 0 订阅

订阅专栏

题目：

给定一个按照升序排列的长度为 n 的整数数组，以及 q个查询。

对于每个查询，返回一个元素 k 的起始位置和终止位置（位置从 00 开始计数）。

如果数组中不存在该元素，则返回 -1 -1。

输入格式

第一行包含整数 n 和 q，表示数组长度和询问个数。

第二行包含 n个整数（均在 1∼100001∼10000 范围内），表示完整数组。

接下来 q 行，每行包含一个整数 k，表示一个询问元素。

输出格式

共 q行，每行包含两个整数，表示所求元素的起始位置和终止位置。

如果数组中不存在该元素，则返回 -1 -1。

数据范围

1≤n≤1000001
1≤q≤10000

1≤k≤10000

输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

二分法思想

1、二分法就是寻找一个边界。该边界的左边满足一个性质，右边满足另外一个性质。同时两个性质是对立的，即如果不满足左边区域的数一定满足右边区域。

2、二分法每次循环能排除约一半的数据量（每次只选择一边继续查找），所以整体的时间复杂度只有O（nlogn）

3、二分法的关键在于左右边性质的确定以及mid值的确定。

图解二分法

前叙：

1、首先看中间的两个箭头，这两个箭头实际上是指我们要确认的边界。这个边界可以是一个具体要找到的一个数，也可以是一个性质的边界。

2、左右两边就是被这个边界分割开的两个区域。mid此时可能落在左右任何一个地方。落在左边和右边的处理方法不同

具体步骤：

1、确认左右区域所对应的性质。例如本题左边是小于x，右边是大于x。

2、确认x边界属于左边还是右边。例如本题找起始位置时，由于右边仍可能有x相同值，所以让x属于右区域。（若让其属于左区域，则两边的范围都将包括x，那么就是非对立了）

3、根据性质，具体去看是变化L，还是变化R。总共两者变化方式1、l=mid，r=mid-1 2、l=mid+1,r=mid

4、如果是l=mid的情况将mid的值改为（l+r+1）/2。如果是l=mid+1则mid值为（l+r）/2

本题代码：

#include<iostream>
using namespace std;

int main(){
    int num[100001]={0};
    int k,q,n=0; //k表示查询得数，q表示要查询多少个数,n表示数组中有多少数
    cin>>n>>q;
    for(int i=0;i<n;i++){
        scanf("%d",&num[i]);
    }
    while(q--){
    //找起始位置
    cin>>k;
    int l=0;
    int r=n-1;
        while(l<r){
            int mid=(l+r)>>1;
            if(num[mid]>=k)r=mid;
            else l=mid+1;
        }
        if(num[l]!=k){
            cout<<"-1 -1"<<endl;
        }
        else{
            cout<<l<<" ";
            //当起始位置存在且找到后，我们再去寻找结束位置
            l=0;r=n-1;
            while(l<r){
                int mid=(l+r+1)>>1;//l=mid决定其要格外加1（当l=r-1时不加1死循环）
                if(num[mid]<=k)l=mid;
                else r=mid-1;
            }
            cout<<l<<endl;
        }
    }
}

二分法背诵模板：

//查找左边界 SearchLeft 简写SL
int SL(int l, int r)
{
    while (l < r)
    {
        int mid = l + r >> 1;
        if (check(mid)) r = mid; 
        else l = mid + 1; 
    }   
    return l;
}
//查找右边界 SearchRight 简写SR 
int SR(int l, int r) 
{
    while (l < r)
    {                   
        int mid = l + r + 1 >> 1; //需要+1 防止死循环
        if (check(mid)) l = mid;
        else r = mid - 1; 
    }
    return r; 
}