Cod sursa(job #3242101)

Utilizator Sasha_12454Eric Paturan Sasha_12454 Data 8 septembrie 2024 21:22:04
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <bits/stdc++.h>

const std :: string FILENAME = "stramosi";

std :: ifstream in (FILENAME + ".in");

std :: ofstream out (FILENAME + ".out");

const int NMAX = 25e4 + 5;

int n;

int m;

int p;

int q;

int parent[NMAX];

int jump[NMAX];

int depth[NMAX];

std :: vector <int> v[NMAX];

void dfs(int nod)
{
    for(int i : v[nod])
    {
        depth[i] = depth[nod] + 1;

        if(depth[nod] - depth[jump[nod]] == depth[jump[nod]] - depth[jump[jump[nod]]])
        {
            jump[i] = jump[jump[nod]];
        }
        else
        {
            jump[i] = nod;
        }

        dfs(i);
    }
}

inline int f(int nod, int stramos)
{
    int d = depth[nod] - stramos;

    if(d < 0)
    {
        return 0;
    }

    while(depth[nod] != d)
    {
        if(depth[jump[nod]] >= d)
        {
            nod = jump[nod];
        }
        else
        {
            nod = parent[nod];
        }
    }

    return nod;
}

int main()
{

    in >> n >> m;

    for(int i = 1; i <= n; i ++)
    {
        in >> parent[i];

        v[parent[i]].push_back(i);
    }

    for(int i = 1; i <= n; i ++)
    {
        if(parent[i] == 0)
        {
            jump[i] = i;

            dfs(i);
        }
    }

    while(m --)
    {
        in >> q >> p;

        out << f(q, p) << '\n';
    }

    return 0;
}