Cod sursa(job #2489351)

Utilizator andreighinea1Ghinea Andrei-Robert andreighinea1 Data 8 noiembrie 2019 17:18:54
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <iostream>
#include <fstream>
#define Nmax 250002

using namespace std;

FILE *f=fopen("stramosi.in","rt");
ofstream o("stramosi.out");

int n,m,t,d[20][Nmax],q,p,i,k,putere; //2^20

int main()
{
    fscanf(f,"%d%d",&n,&m);
    for(i=1;i<=n;++i)
        fscanf(f,"%d",&d[0][i]);

    for(k=1;(1<<k)<=n;++k)
        for(i=1;i<=n;++i)
            d[k][i]=d[k-1][d[k-1][i]]; // d[k][i] == al 2^k -lea stramos al lui i

    for(i=1;i<=m;++i){
        fscanf(f,"%d%d",&q,&p);

        putere=0;
        while(p>0){
            if(p&1)
                q=d[putere][q]; // ne deplasam in sus pe arbore cu 2^(putere)
            ++putere;
            p>>=1; // iau urmatoarea putere a lui 2 (p/=2)
        }
        o << q << '\n';
    }
    return 0;
}