[AcWing]789. 数的范围

目录

一、题目描述：

二、题目分析：

三、题解代码：

---

一、题目描述：

二、题目分析：

比较经典的二分，我们需要用两次二分法来找出询问元素第一次出现的位置和最后一次出现的位置，首先是找询问元素第一次出现的位置，由于该位置的左边元素都小于询问元素，设 mid 为数组的中间元素，t 为询问元素，所以此时首要考虑的就是找出 a[mid] < t 时的情况，此时的 a[mid] 为 t 左边的第一个元素，所以缩短左边界到 mid+1 的位置，即为第一个 t 出现的位置；其次是找出最后一次询问元素出现的位置，由于该位置的右边元素都大于询问元素，所以首要考虑的就是 a[mid]>t 时的情况，此时的 a[mid] 为 t 右边的第一个元素，所以缩短右边界到 mid-1 的位置，也就是最后一个t出现的位置。

若不考虑时间复杂度，第二个位置也可从第一个位置开始遍历数组搜出。 

三、题解代码：

#include<bits/stdc++.h>
using namespace std;

const int N = 1e5+10;
int a[N];
int n,q,t;

int main()
{
    scanf("%d%d",&n,&q);
    
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
        
    while(q--)
    {
        int l=0,r=n-1;
        
        scanf("%d",&t);
        
        while(l<r)//第一次二分
        {
            int mid=(l+r)>>1;
            if(a[mid]<t)//找左边的第一个t
                l=mid+1;//第一个t的位置
            else
                r=mid;
        }
        
        if(a[l] != t)
        {
            cout<<-1<<" "<<-1<<endl;
            continue;
        }
            
        int l1=l,r1=n-1;
        while(l1<r1)//第二次二分
        {
            int mid1=(l1+r1+1)>>1;//若mid可以取到l1,且不+1，会出现数组过短导致死循环的情况
            if(a[mid1]>t)//找右边的第一个t
                r1=mid1-1;//最后一个t的位置
            else
                l1=mid1;
        }
        
        cout<<l<<" "<<r1<<endl;
    }
    
    return 0;
}