Cod sursa(job #2904456)

Utilizator iulia.talpalariuIulia-Georgiana Talpalariu iulia.talpalariu Data 17 mai 2022 23:40:36
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.08 kb
#include <fstream>

int matrice[18][250001];
int main() {
    std::fstream fileIn("stramosi.in");
    std::ofstream fileOut("stramosi.out");
    int n, m;
    fileIn >> n >> m;
    for(int i = 1; i <= n ; ++i) {
        fileIn >> matrice[0][i]; // pe prima linie pun vectorul nostru de stramosi
    }
    //preprocesez matricea cu stramosi din puteri a lui 2 in puteri a lui 2
    for(int i = 1 ; (1 << i) <= n; ++i) {
        for(int j = 1; j <=  n; ++j){
            matrice[i][j] =  matrice[i-1][matrice[i-1][j]];
        }
    }
    //construiesc rasp
    int a,b; // a membrul si al b-lea stramos de gasit
    for(int i = 0; i < m; ++i) {
        fileIn >> a >> b;
        int x;
        x = a; //plec de la membrul dat si voi sari din puteri a lui 2
        while(b) { //cat inca nu am terminat stramosii
            int k = 0;
            while((1<<(k + 1)) <= b) k++;//calculez log afland poz MSB
            x = matrice[k][x];
            b -= (1 << k); // update --peste cati stramosi mai am de sarit
        }
        fileOut << x << '\n';
    }

    return 0;
}