Cod sursa(job #1695367)

Utilizator al.mocanuAlexandru Mocanu al.mocanu Data 26 aprilie 2016 23:00:52
Problema Stramosi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.7 kb
#include <stdio.h>
#include <vector>
using namespace std;

const int MAX = 250005;
int n, m, x, y, p[MAX][25];

int anc(int x, int y){
	int pow = 0;
	while(y && x){
		if(y & 1){
			if(!p[x][pow])
				return 0;
			x = p[x][pow];
		}
		pow++;
		y >>= 1;
	}
	return x;
}

int main(){
	freopen("stramosi.in", "r", stdin);
	freopen("stramosi.out", "w", stdout);
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i++)
		scanf("%d", &p[i][0]);

	int ok = 1, cnt = 0;
	while(ok){
		ok = 0;
		for(int i = 1; i <= n; i++)
			if(p[i][cnt] && p[p[i][cnt]][cnt]){
				ok = 1;
				p[i][cnt + 1] = p[p[i][cnt]][cnt];
			}
		cnt++;
	}
	for(int i = 0; i < m; i++){
		scanf("%d%d", &x, &y);
		printf("%d\n", anc(x, y));
	}
	return 0;
}