Cod sursa(job #3257540)

Utilizator Mihai_OctMihai Octavian Mihai_Oct Data 18 noiembrie 2024 13:14:06
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.68 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("stramosi.in");
ofstream fout("stramosi.out");
const int Max = 19;
int stramos[250002][21];
int n, q, i, t;

static inline int Cb(int nod, int val) {
    for(int i = 0; i <= Max; i++) {
        if(val >> i & 1) nod = stramos[nod][i];
    }
    return nod;
}

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

    for(int b = 1; b <= Max; b++) {
        for(i = 1; i <= n; i++) stramos[i][b] = stramos[stramos[i][b - 1]][b - 1];
    }

    while(q--) {
        int q, p;
        fin >> q >> p;

        fout << Cb(q, p) << "\n";
    }

    return 0;
}