Cod sursa(job #2641500)

Utilizator euyoTukanul euyo Data 11 august 2020 17:51:16
Problema Stramosi Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <fstream>

using namespace std;

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

const int MaxN = 250002;
const int MaxLog = 20;

int anc[MaxN][MaxLog]; //anc[i][p] = indicele stramosului 2^p al lui i
int log[MaxN];

int main() {
  int n, q;

  fin >> n >> q;
  for ( int i = 1; i <= n; ++i ) {
	fin >> anc[i][0];
  }
  for ( int p = 1; p <= MaxLog - 1; ++p ) {
    for	( int i = 1; i <= n; ++i ) {
      anc[i][p] = anc[anc[i][p - 1]][p - 1];
	}
  }
  for ( int i = 2; i <= n; ++i ) {
	log[i] = log[i / 2] + 1;
  }
  while ( q-- ) {
	int p, i;
	fin >> i >> p;
    int lg = log[p];
    int b = i;
	while ( p ) {
	  b = anc[b][lg];
	  p -= (1 << lg);
	  lg = log[p];
	}
	fout << b << "\n";
  }
  fin.close();
  fout.close();
  return 0;
}