Cod sursa(job #3230382)

Utilizator danielbirsannBirsan Daniel danielbirsann Data 20 mai 2024 22:56:24
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <bits/stdc++.h>
#define N 250005
using namespace std;

ifstream fin("stramosi.in");
ofstream fout("stramosi.out");

int n, q;

int d[19][N];

int main() {

    fin >> n >> q;

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

    for (int i = 1; i <= 18; i++) {
        for (int j = 1; j <= n; j++) {
            d[i][j] = d[i-1][d[i-1][j]];
        }
    }
    while (q--) {
        int member, steps;
        fin >> member >> steps;

        int power = 0;
        int result = member;

        while (steps) {
            if (steps % 2 == 1) {
                result = d[power][result];
            }
            power++;
            steps /= 2;
        }

        fout << result << "\n";
    }

    return 0;
}