Cod sursa(job #2976780)

Utilizator indianu_talpa_iuteTisca Catalin indianu_talpa_iute Data 9 februarie 2023 23:30:32
Problema Cautare binara Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.43 kb
#include <bits/stdc++.h>
#define MAXSZ 100000

using namespace std;

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

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

    return arr[hi] == x ? hi + 1 : -1;
}

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

    return hi + 1;
}

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

    return lo + 1;
}

int main() {
    int n, arr[MAXSZ];
    fin >> n;
    for (int i = 0; i < n; i++)
        fin >> arr[i];

    int m;
    fin >> m;
    for (int q = 0; q < m; q++) {
        int t, x;
        fin >> t >> x;
        switch (t) {
            case 0:
                fout << binarySearch(n, arr, x) << '\n';
                break;
            case 1:
                fout << lowerBound(n, arr, x) << '\n';
                break;
            case 2:
                fout << higherBound(n, arr, x) << '\n';
                break;
        }
    }
    return 0;
}