Cod sursa(job #3179771)

Utilizator Vlad_NistorNIstor Vlad Vlad_Nistor Data 4 decembrie 2023 10:33:37
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.04 kb
#include <bits/stdc++.h>
using namespace std;
#define NMAX 250005
#define MAXLOG 19

int tata[NMAX], stramosi[MAXLOG][NMAX],n;

void __initDinamics__(){
    for(int i = 1;i<=n;i++){
        /// stramosul cu numarul 2^0 (1) este de fapt tatal nodului
        stramosi[0][i] = tata[i];
    }
    for(int i = 1;i<MAXLOG;i++){
        for(int j = 1;j<=n;j++){
            stramosi[i][j] = stramosi[i-1][stramosi[i-1][j]];
        }
    }
}
int getStramos(int nod, int k){
    /// determinarea celui de al k lea nod se reduce in scrierea in baza 2 a lui k
    for(int i = 0;i< MAXLOG; i++){
        if(k & (1 << i)){
            nod = stramosi[i][nod];
        }
    }
    return nod;
}

int main(void){
    ofstream cout("stramosi.out");
    ifstream cin("stramosi.in");
    int T;
    cin >> n >> T;
    for(int i= 1;i<=n;i++){
        int x;
        cin >> x;
        tata[i] = x;
    }
    __initDinamics__();
    while(T--){
        int nod, k;
        cin >> nod >> k;
        cout << getStramos(nod,k) << '\n';
    }
}