Cod sursa(job #1787889)

Utilizator CrystyAngelDinu Cristian CrystyAngel Data 25 octombrie 2016 10:45:33
Problema Stramosi Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.66 kb
#include <iostream>
#include <fstream>
#include <cmath>

using namespace std;

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

#define nmax 250100

int rmq[20][nmax];
int n,q,i,j,x,nr,k;

inline int compute(int x,int nr)
{
    while(nr && x)
    {
        int k = log2(nr);
        x = rmq[k][x];
        nr = nr-(1<<k);
    }
    return x;
}

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

    k = log2(n);
    for(j=1; j<=k; ++j)
        for(i=1; i<=n; ++i)
            rmq[j][i]=rmq[j-1][rmq[j-1][i]];

    while(q--)
    {
        fin>>x>>nr;
        fout<<compute(x,nr)<<'\n';
    }
}