Cod sursa(job #3184726)

Utilizator PsyDuck1914Feraru Rares-Serban PsyDuck1914 Data 16 decembrie 2023 17:00:00
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f ("stramosi.in");
ofstream g ("stramosi.out");

const int NMAX = 25e4;
const int LMAX = 19;
int v[NMAX+1];
int str[NMAX+1][LMAX+1];

int query(int q, int p){

    for(int j=LMAX; j>=0; j--)
        if((1 << j) & p)
            q = str[q][j];
    return q;
}

int main()
{
    int n, t;
    f >> n >> t;

    for(int i=1; i<=n; i++)
        f >> v[i], str[i][0] = v[i];

    for(int j=1; j<=LMAX; j++)
        for(int i=1; i<=n; i++)
            str[i][j] = str[str[i][j-1]][j-1];

    for(int i=1; i<=t; i++){
        int q, p;
        f >> q >> p;
        g << query(q, p) << "\n";
    }


    return 0;
}