Cod sursa(job #3289335)

Utilizator stanciuvalentinStanciu-Tivlea Valentin Gabriel stanciuvalentin Data 26 martie 2025 15:58:49
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.64 kb
#include <bits/stdc++.h>

using namespace std;

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

int n,q,dp[250200][22],p,lung;

int main()
{
    f>>n>>q;
    for(int i=1; i<=n; i++)
        f>>dp[i][0];
    for(int j=2; j<=n; j++)
    {
        for(int i=1; i<=20; i++)
            if(dp[j][i-1])
            {
                int p=j, l=i;
                dp[p][l]=dp[dp[p][l-1]][l-1];
            }
    }
    for(int i=1; i<=q; i++)
    {
        f>>p>>lung;
        while(lung)
        {
            int l=log2(lung);
            p=dp[p][l], lung-=(1<<l);
        }
        g<<p<<'\n';
    }
    return 0;
}