《二分查找》- acwing 789.数的范围

题目

1、题目描述

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

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

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

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

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

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

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

  如果数组中不存在该元素，则返回 -1 -1。
数据范围
  1≤ $n$ ≤100000
  1≤ $q$ ≤10000
  1≤ $k$ ≤10000

输入样例：
  6 3
  1 2 2 3 3 4
  3
  4
  5
输出样例：
  3 4
  5 5
  -1 -1

2、基础框架

int erfen_left(int a[], int target,int l, int r){  //左边界
}
int erfen_right(int a[], int target, int l, int r){//右边界
}

3、原题链接

acwing 789.数的范围.

结题报告

1、思路

划分数组，找到左边界和右边界。
找左边界：mid = l + r >> 1,
找右边界：mid = l + r + 1 >>1;
如果a[r] != target; 那么返回 -1;

2、代码

#include <iostream>

using namespace std;
int erfen1(int a[], int b,int l, int r){
    if(a[r]< b || a[l] > b) return -1;  //不存在
  
    while(l < r){
        int  mid = l + r>>1;  //左边界不用加一
        if(a[mid] < b) l = mid+1;
        else r = mid;
    }

    if(a[r] == b)
        return r;
    else //数组不存在target
        return -1;
    
}
int erfen2(int a[], int b, int l, int r){
    if(a[r] < b || a[l]> b) return -1;//数组不存在target
    
    while(l < r){
        int mid = l +r +1>> 1;   //右边界加一
        if(a[mid] <= b) l = mid;
        else r = mid-1;
    }
    
    if(a[r] == b)
        return r;
    else //数组不存在target
        return -1;
}

int main(){
    int m, n;
    cin>>m>>n;
    
    int a[m];
    for(int i = 0; i < m; i++) cin>>a[i];
    int b[n];
    for(int i = 0; i < n; i++) cin>>b[i];
    
    for(int i = 0 ; i < n; i++){
        cout<<erfen1(a, b[i], 0 , m- 1)<<' '<< erfen2(a, b[i], 0 ,m -1)<<endl;
    }
}

注意
 主要是左边界和右边界的查找,搞懂原理就好了。