Cod sursa(job #2126457)

Utilizator B_RazvanBaboiu Razvan B_Razvan Data 9 februarie 2018 17:24:35
Problema Cautare binara Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.58 kb
#include <iostream>
#include <cstdio>
#define NMAX 100005

using namespace std;

int N, v[NMAX], M;

void read()
{
    scanf("%d", &N);
    for(int i=1; i<=N; ++i)
        scanf("%d", &v[i]);
}

int solve0(int x, int biti)
{
    int sol = 0;
    for(int i = biti; i != 0; i >>= 1)
        if(sol + i <= N  && v[sol + i] <= x)
            sol += i;
    if(v[sol] == x)
        return sol;
    return -1;
}

int solve1(int x, int biti)
{
    int sol = 0;
    for(int i = biti; i != 0; i >>= 1)
        if(sol + i <= N && v[sol + i] <= x)
            sol += i;
    return sol;
}

int solve2(int x, int biti)
{
    int sol = N;
    for(int i = biti; i != 0; i >>= 1)
        if(sol - i > 0 && v[sol - i] >= x)
            sol -= i;
    return sol;
}

void readOperations()
{
    int biti;
    for(biti = 1; biti <= N; biti <<= 1);

    scanf("%d", &M);
    while(M)
    {
        int operatie, x;
        scanf("%d %d", &operatie, &x);
        if(operatie == 0)
            printf("%d\n", solve0(x, biti)); /// cea mai mare pozitie pe care se afla un element cu val x sau -1 daca nu exista
        else if(operatie == 1)
            printf("%d\n", solve1(x, biti)); /// cea mai mare pozitie pe care se afla un element cu val mai mica sau egala cu x
        else
            printf("%d\n", solve2(x, biti)); /// cea mai mica pozitie pe care se afla un element cu val mai mare sau egala cu x
        M--;
    }
}

int main()
{
    freopen("cautbin.in", "r", stdin);
    freopen("cautbin.out", "w", stdout);
    read();
    readOperations();
    return 0;
}