Cod sursa(job #2506830)

Utilizator Ioana_GaborGabor Ioana Ioana_Gabor Data 8 decembrie 2019 20:23:22
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb

#include <iostream>
#include <fstream>
#include <vector>
#define NMAX 250000

using namespace std;

ifstream f("stramosi.in");
ofstream g("stramosi.out");

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

int walk(int varf,int dist){
    for(int j=17;j>=0;j--){
        if((1<<j)<=dist){
            varf=table[varf][j];
            dist-=(1<<j);
        }
    }
    return varf;
}

int main()
{
    ios_base::sync_with_stdio(false);
	f.tie(0); g.tie(0);
    f>>n>>m;
    int i,j;
    for(i=1;i<=n;i++){
        f>>table[i][0];
    }
    for(j=1;j<=17;j++){
        for(i=1;i<=n;i++){
            table[i][j]=table[table[i][j-1]][j-1];
        }
    }
    while(m--){
        f>>q>>p;
        g<<walk(q,p)<<'\n';
    }
    f.close();
    g.close();
    return 0;
}