Cod sursa(job #3237950)

Utilizator eZ_tAtDarius Tat eZ_tAt Data 14 iulie 2024 15:36:41
Problema Cautare binara Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.28 kb

#include <vector>
#include <fstream>

using namespace std;

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

int cb0(const vector<int>& a, int n, int x) {
    int st = 0; 
    int dr = n - 1;
    int rez = -1;

    while(st <= dr) {
        int mij = (st + dr) >> 1;
        if(a[mij] <= x) {
            rez = mij;
            st = mij + 1;
        } else {
            dr = mij - 1;
        }
    }
    if (rez == -1 || a[rez] != x) {
        return -1;
    } else {
        return rez + 1;
    }
}

 int cb2(const vector<int>& a, int n, int x) {
    int st = 0; 
    int dr = n - 1;
    int rez = -1;
    
    while(st <= dr) {
        int mij = (st + dr) >> 1;
        if(a[mij] >= x) {
            rez = mij;
            dr = mij - 1;
        } else {
            st = mij + 1;
        }
    }
    if (rez == -1 || a[rez] != x) {
        return -1;
    } else {
        return rez + 1;
    }
}

int main() {
    int n;
    fin >> n;

    vector<int> a(n);
    for(int i = 0; i < n; i++) {
        fin >> a[i];
    }

    int q;
    fin >> q;
    int x;
    short cerr;
    
    while(q--) {
        fin >> cerr >> x;
        if(cerr == 0 || cerr == 1) {
            fout << cb0(a, n, x) << '\n';
        } else {
            fout << cb2(a, n, x) << '\n';
        }
    }
    
    return 0;
}