Cod sursa(job #2875366)

Utilizator indianu_talpa_iuteTisca Catalin indianu_talpa_iute Data 21 martie 2022 14:58:40
Problema Cautare binara Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.3 kb
#include <bits/stdc++.h>
#define MAX_N 100000

using namespace std;

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

int cb2(int n, int arr[], int x) {
    int lo = 0, hi = n - 1, poz;
    while (lo <= hi) {
        int mid = (lo + hi) / 2;
        if (arr[mid] <= x) {
            lo = mid + 1;
            poz = mid;
        } else
            hi = mid - 1;
    }

    return poz;
}

int cb1(int n, int arr[], int x) {
    int poz = cb2(n, arr, x);
    return (arr[poz] == x ? poz : -1);
}

int cb3(int n, int arr[], int x) {
    int lo = 0, hi = n - 1, poz;
    while (lo <= hi) {
        int mid = (lo + hi) / 2;
        if (arr[mid] >= x) {
            hi = mid - 1;
            poz = mid;
        } else
            lo = mid + 1;
    }
    return poz;
}

int main()
{
    int n, m, arr[MAX_N];
    fin >> n;
    for (int i = 0; i < n; i++)
        fin >> arr[i];
    fin >> m;
    for (int i = 0; i < m; i++) {
        int t, x;
        fin >> t >> x;
        switch (t) {
        case 0:
            fout << cb1(n, arr, x) + 1 << '\n';
            break;
        case 1:
            fout << cb2(n, arr, x) + 1 << '\n';
            break;
        case 2:
            fout << cb3(n, arr, x) + 1 << '\n';
            break;
        }
    }
    return 0;
}