Cod sursa(job #2337239)

Utilizator pishcotMiruna Turbatu pishcot Data 6 februarie 2019 08:38:44
Problema Cautare binara Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.91 kb

#include <bits/stdc++.h>

using namespace std;
ifstream fin("cautbin.in");
ofstream fout("cautbin.out");
int n , a[100005];
///cauta cea mai din dreapta pozitie unde se afla x si o returneaza sau
///returneaza -1 daca x nu se afla in vector
int cautbin0(int x)
{
    if(x > a[n]) return -1;
    if(x < a[1]) return -1;
    int st , dr , mij , poz;
    st = 1;
    dr = n;
    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;
}

///cauta si returneaza cea mai din dreapta pozitie cu a[pozitie] <= x
int cautbin1(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;
}

///cauta si returneaza cea mai din stanga pozitie cu a[pozitie] >= x
int cautbin2(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 op ,M , x;
    fin >> n;
    for(int i = 1; i <= n; i++)
        fin >> a[i];
    fin >> M;
    for(int  i = 1; i <= M; i++)
    {
        fin >> op >> x;
        if(op == 0) fout << cautbin0(x) << "\n";
        else if(op == 1) fout << cautbin1(x) << "\n";
        else fout << cautbin2(x) << "\n";

    }

    return 0;
}