Cod sursa(job #797869)

Utilizator psycho21rAbabab psycho21r Data 15 octombrie 2012 07:49:40
Problema Cautare binara Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.26 kb
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
bool comp(int a, int b)
{
    return a <= b;
}
int main()
{
    ifstream in("cautbin.in");
    ofstream out("cautbin.out");
    int N, M;
    in >> N;
    vector<int> elements(N);
    for(int i = 0; i < N; ++i)
        in >> elements[i];
    sort(elements.begin(), elements.end());
    in >> M;
    for(int i = 0, type, query; i < M; ++i)
    {
        in >> type >> query;
        int exact;
        switch(type)
        {
            case 0:
                exact = upper_bound(elements.begin(), elements.end(), query) - elements.begin() - 1;
                if(elements[exact] == query)
                    out << exact + 1 << "\n";
                else
                    out << -1 << "\n";
                break;
            case 1:
                //All start being higher/equal
                out << lower_bound(elements.begin(), elements.end(), query) - elements.begin() << "\n";
                break;
            case 2:
                //This and all before are smaller/equal
                out << upper_bound(elements.begin(), elements.end(), query) - elements.begin() - 1 << "\n";
        }
    }
    in.close();
    out.close();
    return 0;
}