Cod sursa(job #1962857)

Utilizator razvan242Zoltan Razvan-Daniel razvan242 Data 12 aprilie 2017 09:09:34
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <fstream>

using namespace std;

ifstream fin("stramosi.in");
ofstream fout("stramosi.out");

const int NMAX = 250001;
const int LGMAX = 21;

int dp[LGMAX][NMAX];
int n, m;

inline void read() {
    fin >> n >> m;
    for (int i = 1; i <= n; ++i)
        fin >> dp[0][i];

    for (int lg = 1; lg < LGMAX; ++lg)
        for (int i = 1; i <= n; ++i)
            dp[lg][i] = dp[lg - 1][dp[lg - 1][i]];
}

inline int query(int x, int y) {
    for (int lg = 0; (1 << lg) <= y; ++lg) {
        if (y & (1 << lg))
            x = dp[lg][x];
    }
    return x;
}

inline void solve() {
    int x, y;
    while (m--) {
        fin >> x >> y;
        fout << query(x, y) << '\n';
    }
}

int main()
{
    read();
    solve();

    fin.close();
    fout.close();
    return 0;
}