Cod sursa(job #2206871)

Utilizator triscacezarTrisca Vicol Cezar triscacezar Data 24 mai 2018 08:29:17
Problema Stramosi Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <bits/stdc++.h>

using namespace std;
ifstream f("stramosi.in");
ofstream g("stramosi.out");
vector<int> r,v[250010];
int n,m,i,t[40][250010];
void dfs(int poz)
{
    for(auto it:v[poz])
        dfs(it);
    for(int i=1;i<=35;i++)
        t[i][poz]=t[i-1][t[i-1][poz]];
}
int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>t[0][i];
        v[t[0][i]].push_back(i);
        if(!t[0][i])
            r.push_back(i);
    }
    for(auto it:r)
        dfs(it);
    for(;m;m--)
    {
        int x,y,i=0;
        f>>y>>x;
        while(x)
        {
            if(x&1)
                y=t[i][y];
            x>>=1;i++;
        }
        g<<y<<'\n';
    }
    return 0;
}