0
点赞
收藏
分享

微信扫一扫

整数二分模板


文章目录

  • ​​Question​​
  • ​​Ideas​​
  • ​​Code​​
  • ​​C++​​
  • ​​Python​​

Question

给定一个按照升序排列的长度为 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

Ideas

Code

C++

#include <iostream>

using namespace std;
const int N = 1e5 + 10;
int a[N];

int main()
{
int n, q;
scanf("%d%d", &n, &q);

for (int i = 0; i < n; i ++) scanf("%d", &a[i]);

int x;
while(q --)
{
scanf("%d", &x);
int l = 0, r = n - 1;
while(l < r)
{
int mid = l + r >> 1;
if(a[mid] >= x) r = mid;
else l = mid + 1;
}
if (a[l] == x)
{
printf("%d ", l);
int r = n - 1;

while(l < r)
{
int mid = l + r + 1 >> 1;
if (a[mid] <= x) l = mid;
else r = mid - 1;
}
printf("%d\n", r);
}
else
puts("-1 -1");
}

return 0;
}

Python

n, q = list(map(int, input().strip().split()))

a = list(map(int, input().strip().split()))

for i in range(q):
x = int(input())
l = 0
r = n - 1
while l < r:
mid = l + r >> 1
if a[mid] >= x:
r = mid
else:
l = mid + 1
if a[l] != x:
print("-1 -1")
else:
print(f"{l}", end = ' ')
r = n - 1
while l < r:
mid = l + r + 1>> 1
if a[mid] <= x:
l = mid;
else:
r = mid - 1;
print(f"{r}")在这里插入代码片


举报

相关推荐

0 条评论