Cod sursa(job #1377774)

Utilizator sherban26FMI Mateescu Serban-Corneliu sherban26 Data 6 martie 2015 01:30:01
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <iostream>
#include <fstream>

#define NMAX 250010
#define LOGMAX 20

using namespace std;

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

int n, m;
int mat[LOGMAX][NMAX];

int main()
{
    fin >> n >> m;

    //citire
    for (int i = 1; i <= n; ++i)
    {
        fin >> mat[0][i];
    }

    //precalc
    for (int i = 1; (1 << i) <= n; ++i)
    {
        for (int j = 1; j <= n; ++j)
        {
            mat[i][j] = mat[i - 1][mat[i - 1][j]];
        }
    }

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

        for (int i = 0; (1 << i) <= q; ++i)
        {
            if ((1 << i) & q) p = mat[i][p];
        }

        fout << p << "\n";
    }

    return 0;
}