Cod sursa(job #2503225)

Utilizator FrostfireMagirescu Tudor Frostfire Data 2 decembrie 2019 18:57:00
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.64 kb
#include <cstdio>
#define NMAX 250000

using namespace std;

int n, q, dp[NMAX+10][20];

int main()
{
    freopen("stramosi.in", "r", stdin);
    freopen("stramosi.out", "w", stdout);
    scanf("%d%d", &n, &q);
    for(int i=1; i<=n; i++) scanf("%d", &dp[i][0]);
    for(int i=1; i<=18; i++)
        for(int j=1; j<=n; j++) dp[j][i] = dp[dp[j][i-1]][i-1];
    while(q--)
        {   int nod, dist, sol;
            scanf("%d%d", &nod, &dist);
            sol = nod;
            for(int i=0; i<=18 && dist; i++) if((dist & (1 << i))) dist = dist ^ (1 << i), sol = dp[sol][i];
            printf("%d\n", sol);
        }
    return 0;
}