Cod sursa(job #1051202)

Utilizator PsychoAlexAlexandru Buicescu PsychoAlex Data 9 decembrie 2013 20:11:39
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <iostream>
#include <fstream>

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

int n, m;
int vec[21][250001];

int rezolvare(int nod, int pos)
{
    int maxim =17;
    while(pos > 0)
    {
        if((1<<maxim) <= pos)
        {
            nod = vec[maxim][nod];
            pos = pos - (1<<maxim);
        }
        maxim--;
    }
    return nod;
}

void citire()
{
    fin>>n>>m;
    for(int i = 1; i <= n; i++)
    {
        fin>>vec[0][i];
    }

    for(int i = 1; (1<<i) <= n; i++)
    {
        for(int j = 1; j <=n; j++)
        {
            vec[i][j] = vec[i - 1][vec[i - 1][j]];
        }
    }
    int x, y;
    for(int i = 0; i < m; i++)
    {
        fin>>x>>y;
        fout<<rezolvare(x, y)<<'\n';
//        std::cout<<rezolvare(x, y)<<'\n';
    }
}

int main()
{
    citire();
    return 0;
}