Cod sursa(job #2753545)

Utilizator truscalucaLuca Trusca truscaluca Data 23 mai 2021 12:47:32
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <iostream>

const int nMax = 250005;

int t[nMax][20], n, m, nod, k, curr;

using namespace std;

int main() {
    freopen("stramosi.in", "r", stdin);
    freopen("stramosi.out", "w", stdout);

    // Input rapid
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> t[i][0];
    }

    for (int p = 1; (1 << p) <= n; p++) {
        for (int i = 1; i <= n; i++) {
            t[i][p] = t[t[i][p - 1]][p - 1];
        }
    }

    for (int q = 1; q <= m; q++) {
        cin >> nod >> k;

        curr = nod;
        int p = 0;
        while (curr && k) {
            if (k % 2 == 1) {
                curr = t[curr][p];
            }
            p++;
            k /= 2;
        }
        cout << curr << "\n";
    }
    return 0;
}