Cod sursa(job #2705955)

Utilizator AlexGFXMatei Alexandru AlexGFX Data 13 februarie 2021 16:22:45
Problema Cautare binara Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 4.52 kb
#include <iostream>
#include <fstream>

using ull = unsigned long long;
using namespace std;

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

const int NMAX = 10100;

ull numere[NMAX];
int n;
int m;

void citire() {
    fin >> n;
    for(int i = 0; i < n; i++) {
        fin >> numere[i];
        //cout << numere[i] << endl;
    }
}

int tip0iterativ(int st, int dr, ull x) {
    int mid, last = dr + 1;
    while(st <= dr) {
        mid = (st + dr) / 2;
        if(numere[mid] <= x) {
            st = mid + 1;
            last = mid;
        } else dr = mid - 1;
    }

    if(numere[last] == x)
        return last;
    else return -2;
}

int tip1iterativ(int st, int dr, ull x) {
    int mid, last = dr + 1;
    while(st <= dr) {
        mid = (st + dr) / 2;
        if(numere[mid] <= x) {
            st = mid + 1;
            last = mid;
        } else if (numere[mid] > x) {
            dr = mid - 1;
        }
    }

    if(numere[last] <= x)
        return last;

}

int tip2iterativ(int st, int dr, ull x) {
    int mid, last = dr + 1;
    while(st <= dr) {
        mid = (st + dr) / 2;
        if(numere[mid] <= x) {
            st = mid + 1;
            last = mid;
        } else dr = mid - 1;
    }

    if(numere[last] >= x)
        return last;
}

int tip0(ull numar, int stanga, int dreapta) {
    int mid = (stanga + dreapta) / 2;

    if (numere[mid] == numar) {
        if (mid + 1 > n)
        {
            return mid;
        }

        for(int i = mid; i < n; i++) {
            if (numere[mid + 1] != numar) {
                return mid;
            }
            mid++;
        }
    }

    if (stanga == dreapta) {
        return -1;
    }

    if(numere[mid] > numar) {
        dreapta = mid - 1;
        tip0(numar, stanga, dreapta);
    } else {
        stanga = mid + 1;


        tip0(numar, stanga, dreapta);
    }
}

int tip1(ull numar, int stanga, int dreapta) {
    int mid = (stanga + dreapta) / 2;

    if (numere[mid] == numar) {
        if (mid + 1 > n)
        {
            return mid;
        }

        for(int i = mid; i < n; i++) {
            if(i + 1 > n) {
                return i;
            }
            if (numere[i + 1] != numar) {
                return i;
            }
        }
    }

    if (stanga == dreapta) {
        if(numere[mid] > numar) {
            if(mid - 1 >= 0) {
                return mid - 1;
            } else {
                return mid;
            }
        }
        if(numere[mid] < numar) {
            for(int i = mid; i > n; i++) {
                if (i + 1 > n) {
                    return i;
                }
                if(numere[i + 1] > numere[i]) {
                    return i;
                }
            }
        }
    }

    if(numere[mid] > numar) {
        dreapta = mid - 1;
        tip1(numar, stanga, dreapta);
    } else {
        stanga = mid + 1;
        tip1(numar, stanga, dreapta);
    }
}

int tip2(ull numar, int stanga, int dreapta) {
    int mid = (stanga + dreapta) / 2;

    if (numere[mid] == numar) {
        if (mid - 1 > n)
        {
            return mid;
        }

        for(int i = mid; i >= 0; i--) {

            if(i - 1 < 0) {
                return i;
            }

            if (numere[i - 1] != numar) {

                return i;
            }
        }
    }

    if (stanga == dreapta) {
        if(numere[mid] > numar) {

            return mid - 1;
        }
        if(numere[mid] < numar) {
            for(int i = mid; i > n; i++) {
                if (i - 1 > n) {
                    return i;
                }
                if(numere[i + 1] > numere[i]) {
                    return i;
                }
            }
        }
    }

     if(numere[mid] > numar) {
        dreapta = mid - 1;
        tip2(numar, stanga, dreapta);
    } else {
        stanga = mid + 1;
        tip2(numar, stanga, dreapta);
    }
}

int cautbin(short tip, ull numar) {
    if(tip == 0) {
        return tip0iterativ(0, n-1, numar);
    }
    if(tip == 1) {
        return tip1iterativ(0, n-1, numar);
    }
    if(tip == 2) {
        return tip2iterativ(0, n-1, numar);
    }
}

void rezolvare() {
    fin >> m;
    for(int i = 0; i < m; i++) {
        short tip;
        ull numar;
        fin >> tip;
        fin >> numar;
        fout << cautbin(tip, numar) + 1 << endl;
    }
}



int main() {

    citire();
    rezolvare();

    return 0;
}