Cod sursa(job #2707344)

Utilizator hhhhhhhAndrei Boaca hhhhhhh Data 16 februarie 2021 20:17:21
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <bits/stdc++.h>
#pragma GCC optimize("O3")
using namespace std;
ifstream fin("stramosi.in");
ofstream fout("stramosi.out");
int n,m,dp[21][250005];
int nod,val,a,i,j;
int query()
{
    int p=nod;
    for(int i=0;i<=18&&val>0;i++)
        if((val>>i)&1)
        {
            p=dp[i][p];
            val-=(1LL<<i);
        }
    return p;
}
int main()
{
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>a;
        dp[0][i]=a;
    }
    for(i=1;i<=18;i++)
        for(j=1;j<=n;j++)
            dp[i][j]=dp[i-1][dp[i-1][j]];
    while(m--)
    {
        fin>>nod>>val;
        fout<<query()<<'\n';
    }
    return 0;
}