Cod sursa(job #2552916)

Utilizator KataIsache Catalina Kata Data 21 februarie 2020 12:30:26
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.64 kb
#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 250000

using namespace std;

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

int n,m,mat[20][NMAX+5],q,p;

int main()
{
    int i,j;
    fin>>n>>m;
    for(i=1;i<=n;i++){
        fin>>mat[0][i];
    }
    for(j=1;j<=17;j++){
        for(i=1;i<=n;i++){
            mat[j][i]=mat[j-1][mat[j-1][i]];
        }
    }
    while(m--){
        fin>>q>>p;
        for( j=17;j>=0;j--){
            if((1<<j)<=p){
                q=mat[j][q];
                p=p-(1<<j);
            }
        }
        fout<<q<<'\n';
    }

    return 0;
}