Cod sursa(job #2226717)

Utilizator inquisitorAnders inquisitor Data 30 iulie 2018 15:24:50
Problema Cautare binara Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.17 kb
#include <cstdio>
#include <algorithm>
#define BUFFER_SIZE 4000000

char buffer[BUFFER_SIZE]; int p = -1;

__attribute__((always_inline)) int get_number()
{
    int nr = 0, c = buffer[++p];

    for(; c < 48 | c > 57; c = buffer[++p]);

    for(; c > 47; nr = nr * 10 + c - 48, c = buffer[++p]);

    return nr;
}

char outBuffer[700000], *writeHead = outBuffer;

__attribute__((always_inline)) void put_int(int x)
{
    if(x == 0)
    {
        *(writeHead++) = 48; *(writeHead++) = 10;

        return;
    }

    if(x < 0)
    {
        *(writeHead++) = 45; *(writeHead++) = 49; *(writeHead++) = 10;

        return;
    }

    char *old = writeHead;

    while(x)
    {
        *(writeHead++) = x % 10 + 48;

        x /= 10;
    }

    std::reverse(old, writeHead);

    *(writeHead++) = 10;
}


int main()
{
    int N, M, v[100005], task, value, log2N, step, i;

    freopen("cautbin.in", "r", stdin);
    freopen("cautbin.out", "w", stdout);

    fread(buffer, 1, BUFFER_SIZE, stdin);

    N = get_number(); ///scanf("%d", &N);

    for(i = 1; i <= N; ++i)
    {
        v[i] = get_number();

        ///scanf("%d", &v[i]);
    }

    for(log2N = 1; log2N <= N; log2N <<= 1);

    M = get_number(); ///scanf("%d", &M);

    for(; M; --M)
    {
        task = get_number();

        value = get_number(); ///scanf("%d %d", &task, &value);

        if(task < 2)
        {
            for(step = log2N, i = 0; step; step >>= 1)
            {
                if(i + step <= N && v[i + step] <= value)
                {
                    i += step;
                }
            }

            if(!task && v[i] != value)
            {
                put_int(-1);
            }
            else
            {
                put_int(i);
            }
        }
        else
        {
            for(step = log2N, i = N; step; step >>= 1)
            {
                if(i - step > 0 && v[i - step] >= value)
                {
                    i -= step;
                }
            }

            put_int(i);
        }
    }

    fwrite(outBuffer, 1, writeHead - outBuffer, stdout);

    return 0;
}