Cod sursa(job #2707340)

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