Cod sursa(job #3312047)

Utilizator Luca_georgescuLuca Georgescu Luca_georgescu Data 25 septembrie 2025 17:58:00
Problema Stramosi Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.65 kb

#include <bits/stdc++.h>

using namespace std;

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

const int nmax=250005;
int n,q,logg=20;
int dp[nmax][35];

int findkth(int x, int k)
{
    for (int i=19; i>=0; i++ )
        if ( k&(1<<i) )
            x=dp[x][i];

    return x;
}

int main()
{
    f >> n >> q;

    for (int i=1; i<=n; i++ )
    {
        int x; f >> x;
        dp[i][0]=x;
    }

    for (int j=1; j<=19; j++ )
       for (int i=1; i<=n; i++ )
            dp[i][j]=dp[dp[i][j-1]][j-1];

    while ( q-- )
    {
        int x,k; f >> x >> k;
        g << findkth(x,k) << '\n';
    }
    return 0;
}