Cod sursa(job #2429362)

Utilizator IulianOleniucIulian Oleniuc IulianOleniuc Data 9 iunie 2019 12:54:02
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <fstream>
#define NMAX 250010

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

int n, q;
int dp[NMAX][19];

int main() {
    fin >> n >> q;
    for (int i = 1; i <= n; i++)
        fin >> dp[i][0];

    for (int j = 1; (1 << j) <= n; j++)
        for (int i = 1; i <= n; i++)
            dp[i][j] = dp[dp[i][j - 1]][j - 1];

    for (int it = 0; it < q; it++) {
        int q, p; fin >> q >> p;
        int ancestor = q;
        for (int i = 18; i >= 0; i--)
            if (p & (1 << i))
                ancestor = dp[ancestor][i];
        fout << ancestor << '\n';
    }

    fout.close();
    return 0;
}