Cod sursa(job #2283133)

Utilizator TooHappyMarchitan Teodor TooHappy Data 15 noiembrie 2018 00:25:51
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 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; scanf("%d %d ", &n, &m);
    // cin >> n >> m;

    for(int i = 1; i <= n; ++i) {
        // cin >> stramos[i][0];
        scanf("%d ", &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; scanf("%d %d ", &q, &p);
        // cin >> q >> p;

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

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

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