Cod sursa(job #2283130)

Utilizator TooHappyMarchitan Teodor TooHappy Data 15 noiembrie 2018 00:21:53
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <bits/stdc++.h>
 
using namespace std;
 
// ifstream in("stramosi.in");
// ofstream out("stramosi.out");

const int Nmax = 250010;

int stramos[Nmax][20]; // stramosul lui i de ordin 2^j

int getStramos(int q, int p) {
    while(p) {
        int k = log2(p);
        q = stramos[q][k];
        p -= (1 << k);
    }

    return q;
}

int main() {
    freopen("stramosi.in", "r", stdin);
    freopen("stramosi.out", "w", stdout);
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);

    int n, m; cin >> n >> m;

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

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

    for(int i = 1; i <= m; ++i) {
        int q, p; cin >> q >> p;

        cout << getStramos(q, p) << "\n";
    }

    // in.close(); out.close();
 
    return 0;
}