Cod sursa(job #3218955)

Utilizator MerlinTheWizardMelvin Abibula MerlinTheWizard Data 28 martie 2024 22:27:14
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <bits/stdc++.h>
#pragma GCC optimize("O3")

using namespace std;

const int NMAX = 25e4 + 5, DMAX = 18;
int n, q, parent[NMAX][DMAX];

int main()
{
    freopen("stramosi.in", "r", stdin);
    freopen("stramosi.out", "w", stdout);

    ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);

    cin >> n >> q;
    for(int i = 1; i <= n; i++)
        cin >> parent[i][0];

    for(int i = 1; i <= DMAX - 1; i++)
        for(int j = 1; j <= n; j++) 
            parent[j][i] = parent[parent[j][i - 1]][i - 1];

    int a, b;
    for(int i = 1; i <= q; i++)
    {
        cin >> a >> b;
        for(int j = 0; j <= DMAX - 1; j++)
        {
            if(b & (1 << j))
                a = parent[a][j];
        }

        cout << a << "\n";
    }
}