Cod sursa(job #1697113)

Utilizator llalexandruLungu Alexandru Ioan llalexandru Data 30 aprilie 2016 19:32:52
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <fstream>
#include <vector>
#define NM 300005

using namespace std;

ifstream fin("stramosi.in");
ofstream fout("stramosi.out");

vector <int> G[NM];

int M[20][NM], n, m;

int c, lv, cpre, pre;

void DFS(int x)
{
    int i;
    c=1;
    M[0][x]=pre;
    for (i=2; i<=lv; i=i*2)
    {
        M[c][x]=M[c-1][M[c-1][x]];
        c++;
    }
    for (i=0; i<G[x].size(); i++)
    {
        lv++;
        cpre=pre;
        pre=x;
        DFS(G[x][i]);
        pre=cpre;
        lv--;
    }
}

int main()
{
    int i, x, j, c, y;
    fin>>n>>m;
    for (i=1; i<=n; i++)
    {
        fin>>x;
        G[x].push_back(i);
    }
    DFS(0);
    for (i=1; i<=m; i++)
    {
        fin>>x>>y;
        c=0;
        if (y>n)
        {
            fout<<0<<'\n';
            continue;
        }
        while (y>0)
        {
            if (y%2==1)
            {
                x=M[c][x];
            }
            c++;
            y=y/2;
        }
        fout<<x<<'\n';
    }
    return 0;
}