Cod sursa(job #1741695)

Utilizator TeodorCotetCotet Teodor TeodorCotet Data 14 august 2016 19:19:06
Problema Stramosi Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <bits/stdc++.h>

using namespace std;

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

const int NMAX = 25e4 + 5;

int n; int m;

vector<int> g[NMAX];

int ancestor[NMAX][19];
int lg[NMAX];

int main() {

	fin >> n >> m;
	for(int i = 1; i <= n ; ++i) {
		int x; fin >> x;
		ancestor[i][0] = x;
	}

	lg[1] = 0;
	lg[0] = 0;

	for(int i = 2; i <= n ; ++i)
		lg[i] = lg[i / 2] + 1;

	for(int i = 1; i <= lg[n]; ++i)
		for(int j = 1; j <= n; ++j)
			ancestor[j][i] = ancestor[ ancestor[j][i - 1] ][i - 1];

	while(m--) {
		int p; int q;
		//al p-lea stramos al lui q
		fin >> q >> p;
		if(p > n) {
			fout << 0 << '\n';
			continue;
		}

		while(q > 0 && p > 0) {
			q = ancestor[q][lg[p]];
			p -= (1 << lg[p]);
		}

		fout << q << '\n';
	}

	return 0;
}