Cod sursa(job #2977235)

Utilizator Iordache_CezarIordache Cezar Iordache_Cezar Data 11 februarie 2023 09:48:19
Problema Stramosi Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <bits/stdc++.h>
#define NMAX 250008

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

int n, m, stramos[NMAX][20], nivel[NMAX];
vector <int> G[NMAX];

void DFS(int nod);
int query(int nod, int p);

int main()
{
    int x, y;
    fin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        fin >> x;
        G[x].push_back(i);
        stramos[i][0] = x;
    }
    for (int j = 1; (1 << j) <= n; j++)
        for (int i = 1; i <= n; i++)
            stramos[i][j] = stramos[stramos[i][j-1]][j-1];

    for (int i = 1; i <= m; i++)
    {
        fin >> x >> y;
        fout << query(x, y) << '\n';
    }
    return 0;
}

int query(int nod, int p)
{
    if (p == 0)
        return nod;
    int x = (int)(log2(p));
    query(stramos[nod][x], p-(1<<x));
}