Cod sursa(job #2185818)

Utilizator savigunFeleaga Dragos-George savigun Data 24 martie 2018 22:15:09
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <iostream>
#include <fstream>
using namespace std;

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

const int MAXN = 250005;
int n, m, father[MAXN], doi[18];
int dp[MAXN][18];

void preprocess() {
    doi[0] = 1;
    for (int i = 1; i <= 17; ++i) doi[i] = doi[i - 1] * 2;

    for (int i = 1; i <= n; ++i) dp[i][0] = father[i];
    for (int j = 1; doi[j] <= n; ++j) {
        for (int i = 1; i <= n; ++i) {
            dp[i][j] = dp[dp[i][j - 1]][j - 1];
        }
    }
}

int query(int x, int k) {
    int pas = 17;
    while (pas >= 0) {
        if (doi[pas] <= k) {
            x = dp[x][pas];
            if (x == 0) return 0;
            k -= (1 << pas);
        }
        pas--;
    }
    return x;
}

int main()
{
    in >> n >> m;
    for (int i = 1; i <= n; ++i) in >> father[i];
    preprocess();

    for (int i = 1; i <= m; ++i) {
        int x, k;
        in >> x >> k;
        out << query(x, k) << "\n";
    }

    return 0;
}