Cod sursa(job #2581205)

Utilizator oldatlantianSerban Cercelescu oldatlantian Data 14 martie 2020 17:27:21
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <bits/stdc++.h>
using namespace std;

ifstream fi("stramosi.in");
ofstream fo("stramosi.out");

const int N = 250005, LG = 19;

vector<int> g[N];
int far[N][LG], lvl[N];

int n, q;

void make_far() {
    for (int lg = 1; lg < LG; ++lg)
    for (int i = 1; i <= n; ++i)
        far[i][lg] = far[far[i][lg - 1]][lg - 1];
}

int get_far(int nod, int lvl) {
    if (lvl > (1 << LG - 1))
        return 0;
    for (int i = 0; i < LG; ++i)
        if ((1 << i) & lvl)
            nod = far[nod][i];
    return nod;
}

int main() {
    fi >> n >> q;
    for (int i = 1; i <= n; ++i)
        fi >> far[i][0];
    make_far();
    
    while (q--) {
        int nod, lvl;
        fi >> nod >> lvl;
        fo << get_far(nod, lvl) << '\n';
    }

    return 0;
}