Cod sursa(job #3232527)

Utilizator pascarualexPascaru Alexandru pascarualex Data 30 mai 2024 18:40:46
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <fstream>
#include <cmath>

int main(){
    std::ifstream fin("stramosi.in");
    std::ofstream fout("stramosi.out");

    int n, m;
    int t[250005], s[250005][25];

    fin >> n >> m;
    for(int i = 1; i <= n; ++i) {

        int x;

        fin >> x;
        t[i] = x;
    }

    for(int i = 1; i <= n; ++i) {
        s[i][0] = t[i];
    }

    for(int i = 0; i <= n; ++i) {
        for(int j = 1; j <= int(std::log2(n)); ++j) {
            s[i][j] = s[s[i][j - 1]][j - 1];
        }
    }


    for(int i = 1; i <= m; ++i) {

        int a, b;

        fin >> a >> b;
        for(int k = 0; k <= int(std::log2(n)); ++k) {
            if (b & (1 << k)) {
                a = s[a][k];
            }
        }

        fout << a << "\n";
    }

    return 0;
}