zl程序教程

您现在的位置是:首页 >  其他

当前栏目

整数二分模板

模板 整数 二分
2023-09-14 09:14:57 时间

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}")在这里插入代码片