Cod sursa(job #2337238)

Utilizator NopeCarp Rafael Nope Data 6 februarie 2019 08:38:09
Problema Cautare binara Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.9 kb
#include <bits/stdc++.h>
using namespace std;

ifstream fin("cautbin.in");
ofstream fout("cautbin.out");

int a[100005] , n;

/// returneaza cea mai din dreapta pozitie unde se afla x
/// sau ret. -1 daca x nu apare in vector
int CB0(int x)
{
    if (x > a[n]) return -1;
    if (x < a[1]) return -1;
    int mij , st = 1 , dr = n , poz;
    poz = -1; /// pozitia cea mai din dreapta unde este x
    while (st <= dr)
    {
        mij = (st + dr) / 2;
        if (a[mij] == x)
        {
            poz = mij;
            /// caut si mai in dreapta un x
            st = mij + 1;
        }
        else if (a[mij] < x) st = mij + 1;
        else dr = mij - 1;
    }
    return poz;
}

/// returneaza cea mai din dreapta pozitie cu a[poz] <= x
int CB1(int x)
{
    if (x < a[1]) return 1;
    if (x >= a[n]) return n;
    int st , dr , mij , poz;
    st = 1;
    dr = n;
    poz = -1;
    while (st <= dr)
    {
        mij = (st + dr) / 2;
        if (a[mij] <= x)
        {
            poz = mij;
            st = mij + 1;
        }
        else
            dr = mij - 1;
    }
    return poz;
}

/// returneaza cea mai din stanga pozitie cu a[poz] >= x
int CB2(int x)
{
    if (x <= a[1]) return 1;
    int st , dr , mij , poz;
    st = 1;
    dr = n;
    poz = -1;
    while (st <= dr)
    {
        mij = (st + dr) / 2;
        if (a[mij] >= x)
        {
            poz = mij;
            dr = mij - 1;
        }
        else
            st = mij + 1;
    }
    return poz;
}

int main()
{
    int i , k , op , x;
    fin >> n;
    for (i = 1 ; i <= n ; i++)
        fin >> a[i];
    fin >> k;
    for (i = 1 ; i <= k ; i++)
    {
        fin >> op >> x;;
        if (op == 0)
            fout << CB0(x) << "\n";
        else if (op == 1)
            fout << CB1(x) << "\n";
        else
            fout << CB2(x) << "\n";
    }
    return 0;
}