Cod sursa(job #2715571)

Utilizator Mihai145Oprea Mihai Adrian Mihai145 Data 3 martie 2021 21:13:58
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <fstream>
#include <vector>

using namespace std;

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

const int NMAX = 250000;
const int LGMAX = 18;

int N, M;
int anc[NMAX + 2][LGMAX + 2];

int main() {
    cin >> N >> M;

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

    for(int e = 1; e <= LGMAX; e++) {
        for(int j = 1; j <= N; j++) {
            anc[j][e] = anc[anc[j][e - 1]][e - 1];
        }
    }

    for(int i = 1; i <= M; i++) {
        int Q, P;
        cin >> Q >> P;

        int res = Q;

        for(int step = LGMAX; step >= 0; step--) {
            if(P >= (1 << step)) {
                res = anc[res][step];
                P -= (1 << step);
            }
        }

        cout << res << '\n';
    }

    return 0;
}