二分查找的关键:有序数组

题目1:数的范围

给定一个按照升序排列的长度为n的整数数组,以及q个查询。对于每个查询,返回一个元素k的起始位置和终止位置(位置从0开始计数)。如果数组中不存在该元素,则返回-1,-1

输入格式

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

第二行包含n个整数,表示完整数组。

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

输出格式

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

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

输入样例:

6 3
1 2 2 3 3 4
3
4
5

输出样例:

3 4
5 5
-1 -1

 

代码

 

//整数二分

#include<iostream>
#include<vector>
using namespace std;

//查找区间左端点
int func_left(vector<int>& vec, int key)
{
  int start = 0;
  int end = vec.size() - 1;
  while (start < end)
  {
    int mid = (start + end) / 2;
    if (vec[mid]>=key)
    {
      end = mid;
    }
    else
    {
      start = mid + 1;
    }
  }
  if (vec[start] != key)
  {
    return -1;
  }
  return start;
}

//查找区间右端点
int func_right(vector<int>& vec, int key)
{
  int start = 0;
  int end = vec.size() - 1;
  while (start < end)
  {
    int mid = (start + end + 1) / 2;
    if (vec[mid]<=key)
    {
      start = mid;
    }
    else
    {
      end = mid-1;
    }
  }
  if (vec[start] != key)
  {
    return -1;
  }
  return end;
}


int main()
{
  int n, p;
  cin >> n;
  cin >> p;
  vector<int> vec1;
  vector<int> vec2;
  for (int i = 0; i < n; i++)
  {
    int tmp;
    cin >> tmp;
    vec1.push_back(tmp);
  }
  for (int i = 0; i < p; i++)
  {
    int tmp;
    cin >> tmp;
    vec2.push_back(tmp);
  }
  for (int i = 0; i < p; i++)
  {
    cout << func_left(vec1,vec2[i]) << " " << func_right(vec1, vec2[i]) << endl;
  }
  return 0;
}

 

posted on 2022-09-20 09:07  yc-limitless  阅读(30)  评论(0)    收藏  举报