Cod sursa(job #2580215)

Utilizator ivan.tudorIvan Tudor ivan.tudor Data 13 martie 2020 13:50:43
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.7 kb
#include<bits/stdc++.h>
using namespace std;
const int N=250005;
int p[N][20];
int lg2[N];
int main()
{
  FILE*fin,*fout;
  fin=fopen("stramosi.in","r");
  fout=fopen("stramosi.out","w");
  int n,q;
  fscanf(fin,"%d%d",&n,&q);
  for(int i=1;i<=n;i++){
    fscanf(fin,"%d",&p[i][0]);
  }
  lg2[1]=0;
  for(int i=2;i<N;i++)
    lg2[i]=lg2[i/2]+1;
  for(int log=1;log<20;log++){
    for(int i=1;i<=n;i++){
      p[i][log]=p[p[i][log-1]][log-1];
    }
  }
  for(int i=1;i<=q;i++){
    int poz,rk;
    fscanf(fin,"%d%d",&poz,&rk);
    while(rk){
      poz=p[poz][lg2[rk]];
      if(poz==0)
        break;
      rk-=(1<<lg2[rk]);
    }
    fprintf(fout,"%d\n",poz);
  }
  return 0;
}