Cod sursa(job #2431615)

Utilizator IulianOleniucIulian Oleniuc IulianOleniuc Data 20 iunie 2019 13:51:51
Problema Stramosi Scor 100
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[19][NMAX];

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

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

    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[i][ancestor];
        fout << ancestor << '\n';
    }

    fout.close();
    return 0;
}