Rango de números C ++ (búsqueda binaria)

Dada una matriz entera de longitud n organizada en orden ascendente y q consultas.
Para cada consulta, devuelve la posición inicial y la posición final de un elemento k (la posición comienza a contar desde 0).
Si el elemento no existe en la matriz, se devuelve "-1 -1".
Formato de entrada La
primera línea contiene números enteros n y q, que indican la longitud de la matriz y el número de consultas.
La segunda línea contiene n números enteros (todos en el rango de 1 a 10000), lo que representa una matriz completa.
Después de q líneas, cada línea contiene un número entero k, que representa un elemento de consulta.
Formato de salida
Un total de q líneas, cada línea contiene dos enteros, indicando la posición inicial y la posición final del elemento a buscar.
Si el elemento no existe en la matriz, se devuelve "-1 -1".
Rango de datos
1≤n≤100000
1≤q≤10000
1≤k≤10000
Ejemplo de entrada:
6 3
1 2 2 3 3 4
3
4
5
Ejemplo de salida:
3 4
5
5-1-1

Código AC:

#include<stdio.h>
#include<algorithm>

int a[100010];
int n,q;
int head,tail;

void binary(int x)
{
    
    
    int l=-1,r=n;
    int middle;
    head=n-1;
    tail=0;

    while(l+1!=r)//找第一个大于等于x的位置
    {
    
    
        //求最小,向下取整
        middle=l+((r-l)>>1);
        if(a[middle]<x){
    
    
            l=middle;
        }
        else if(a[middle]>=x){
    
    
            r=middle;
            if(a[middle]==x) head=std::min(head,middle);
        }
    }
    if(a[head]==x){
    
    //x存在则继续找第一个小于等于x的位置

        l=middle-1;
        r=n;
        while(l+1!=r)
        {
    
    
            //求最大,向上取整
            middle=l+((r-l+1)>>1);
            if(a[middle]<=x){
    
    
                l=middle;
                if(a[middle]==x) tail=std::max(tail,middle);
            }
            else if(a[middle]>x){
    
    
                r=middle;
            }
        }
    }
    else {
    
    //x不存在
        head=-1;
        tail=-1;
    }
}

int main()
{
    
    
    scanf("%d%d",&n,&q);
    for(int i=0;i<n;++i)
    {
    
    
        scanf("%d",&a[i]);
    }
    for(int i=0;i<q;++i)
    {
    
    
        int x;
        scanf("%d",&x);
        binary(x);
        printf("%d %d\n",head,tail);
    }
    return 0;
}

Supongo que te gusta

Origin blog.csdn.net/qq_44643644/article/details/108771110
Recomendado
Clasificación