Cod sursa(job #1758353)

Utilizator elffikkVasile Ermicioi elffikk Data 17 septembrie 2016 02:46:38
Problema Cautare binara Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.8 kb
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;

vector<int> a;

int binMaxEq(int value) {
    int left = 0, right = a.size()-1;
    while (left+1 < right) {
        int middle = (left + right)/2;
        if (a[middle]<=value) {
            left = middle;
        } else {
            right = middle;
        }
    }
    if (value == a[right]) {
        return right;
    }
    if (value == a[left]) {
        return left;
    }
    return -1;
}

int binMaxEqLt(int value) {
    int left = 0, right = a.size()-1;
    while (left+1 < right) {
        int middle = (left + right)/2;
        if (a[middle]<=value) {
            left = middle;
        } else {
            right = middle;
        }
    }
    if (value >= a[right]) {
        return right;
    }
    return left;
}

int binMaxEqGt(int value) {
    int left = 0, right = a.size()-1;
    while (left+1 < right) {
        int middle = left + (right-left)/2;
        if (a[middle]<value) {
            left = middle;
        } else {
            right = middle;
        }
    }
    if (value <= a[left]) {
        return left;
    }
    return right;
}

int main() {
    ifstream cin("cautbin.in");
    ofstream cout("cautbin.out");
    int n, m, x, op;

    cin>>n;
    for (int i=0; i < n; i++) {
        cin>>x;
        a.push_back(x);
    }
    sort(a.begin(), a.end());
    cin>>m;
    for (int i = 0; i < m; i++) {
        cin>>op>>x;
        int v = -1;
        if (op == 0) {
            v = binMaxEq(x);
        } else if (op == 1) {
            v = binMaxEqLt(x);
        } else {
            v = binMaxEqGt(x);
        }
        if ( v < 1) {
            cout << -1 << endl;
        } else {
            cout << v + 1 << endl;
        }
    }
    return 0;
}