Cod sursa(job #2269371)

Utilizator nicolaefilatNicolae Filat nicolaefilat Data 25 octombrie 2018 21:56:30
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <iostream>
#include <fstream>

const int MAXN = 250000 + 5,MAXLOG = 17 + 5;

using namespace std;

ifstream in("stramosi.in");
ofstream out("stramosi.out");

int dp[MAXN][MAXLOG],log[MAXN];
int n,q,i,put;

int query(int nod,int stramos){
    int ans = nod;
    while(stramos){
        ans = dp[ans][log[stramos]];
        if(!ans)
            return 0;
        stramos -= (1 << log[stramos]);
    }
    return ans;
}
int main()
{


    in>>n>>q;
    for(i = 1; i <= n; ++i)
        in>>dp[i][0];
    for(i = 2; i <= n; ++i)
        log[i] = log[i >> 1] + 1;
    for(put = 1; put < MAXLOG; ++put)
        for(i = 1; i <= n; ++i)
            dp[i][put] = dp[dp[i][put - 1]][put - 1];
    for(i = 1; i <= q; ++i){
        int nod,k_stramos;
        in>>nod>>k_stramos;
        out<<query(nod,k_stramos)<<'\n';
    }

    return 0;
}