【藍橋杯練習--二分】789. 數的範圍

二分的時間複雜度是O(log n)
789. 數的範圍
本題使用遍歷查詢會超時，應使用二分

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <math.h>

using namespace std;

int main()
{
    //讀取資料
    int n, q;
    cin >> n >> q;
    
    int arr[n];
    int tar[q];
    for(int i=0; i<n; i++)  cin >> arr[i];
    for(int i=0;i<q;i++)    scanf("%d", &tar[i]);
        
    //二分模版
    for(int i=0; i<q; i++)  //遍歷q次
    {
        //右端點 第一個比tar大的數
        int l=0, r=n-1;
        while(l<r)
        {
            int mid = l + r >> 1;
            if(arr[mid]>=tar[i]) r=mid; 
            else l=mid+1;
        }
        int a=0;
        if(arr[l]==tar[i]) a=l;
        else a=-1;
        
        l=0, r=n-1;
        while(l<r)
        {
            int mid = l + r + 1 >> 1;
            if(arr[mid]<=tar[i]) l=mid; 
            else r=mid-1;
        }
        int b=0;
        if(arr[l]==tar[i]) b=l;
        else b=-1;
        
        cout << a << " " << b << endl;
    }
    
    return 0;
}