Cod sursa(job #2707342)

Utilizator hhhhhhhAndrei Boaca hhhhhhh Data 16 februarie 2021 20:16:16
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.76 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 nod,int val)
{
    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()
{
    ios_base::sync_with_stdio(false);
    fin.tie(0);
    fout.tie(0);
    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--)
    {
        nod,val;
        fin>>nod>>val;
        fout<<query(nod,val)<<'\n';
    }
    return 0;
}