Cod sursa(job #2645615)

Utilizator ursu2001Ursu Ianis-Vlad ursu2001 Data 29 august 2020 10:51:14
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.7 kb
#include <bits/stdc++.h>

using namespace std;

const int MAX_N = 250000;
const int MAX_K = log2(MAX_N) + 1;

int T[MAX_K][MAX_N];

int N, M;

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

    cin >> N >> M;

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

    for(int k = 1; k < MAX_K; ++k)
        for(int i = 1; i <= N; ++i)
            T[k][i] = T[k - 1][T[k - 1][i]];

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

        int k = 0;

        while(P)
        {
            if(P & 1) Q = T[k][Q];

            ++k;
            P = P >> 1;
        }

        cout << Q << '\n';

    }
}