Cod sursa(job #1966494)

Utilizator FckingSlayerSlayer99 FckingSlayer Data 15 aprilie 2017 12:20:58
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <fstream>
#include <string.h>
#include <vector>
#include <algorithm>

#define nMax 250001
#define bit(i) i&(-i)
#define x first
#define y second


using namespace std;

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

int n, nrQuery;
int TT[20][nMax];

int main()
{
    int node, upLev;
    fin>>n>>nrQuery;
    for(int i=1; i<=n; i++)
        fin>>TT[0][i];
    for(int len=1; len<20; len++)
        for(int i=1; i<=n; i++)
            TT[len][i]=TT[len-1][TT[len-1][i]];
    for(int i=1; i<=nrQuery; i++)
    {
        fin>>node>>upLev;
        for(int len=19; len>=0; len--)
        {
            if(upLev>=(1 << len))
            {
                upLev-=(1 << len);
                node=TT[len][node];
            }
        }
        fout<<node<<'\n';
    }
}