Cod sursa(job #2283132)

Utilizator TooHappyMarchitan Teodor TooHappy Data 15 noiembrie 2018 00:24:12
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 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;

        for(int i = 0; (1 << i) <= n && q; ++i) {
            if((p >> i) % 2 == 1) {
                q = stramos[q][i];
            }
        }
        cout << q << "\n";

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

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