Cod sursa(job #374616)

Utilizator cvicentiuCiorbaru Vicentiu Marian cvicentiu Data 17 decembrie 2009 15:39:37
Problema Lowest Common Ancestor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.98 kb
#include <fstream>
#include <vector>
#define MaxN 100528
#define MaxL 100

using namespace std;

fstream fin ("lca.in",ios::in);
fstream fout("lca.out",ios::out);

struct nod {
	int x;
	nod *urm;
};

int n, m, viz[MaxN], k, parc[MaxN<<2], dep[MaxN<<2];
nod *L[MaxN];
int first[MaxN], Lg[MaxN<<2 + 3], M[MaxL][MaxN];

void rmq(){
	for(int i = 2; i <= k; ++i)
		Lg[i] = Lg[i >> 1] + 1;
	for(int i = 1; i <= k; ++i)
		M[0][i] = i;

	for(int i = 1; (1 << i) < k; ++i)
		for(int j = 1; j <= k - (1 << i); ++j)
		{
			int l = 1 << (i - 1);
			M[i][j] = M[i-1][j];
			if(dep[M[i-1][j + l]] < dep[M[i][j]])
				M[i][j] = M[i-1][j + l];
}
	
};

void add_nod(int ex1, int ex2){

	nod *p = new nod;
	nod *q = new nod;

	p->x = ex2;
	p->urm = L[ex1];
	L[ex1] = p;

	q->x = ex1;
	q->urm = L[ex2];
	L[ex2] = q;
};

void euler_det(int x, int depth){ //dfs euler

		viz[x] = 1;
		parc[++k] = x;
		dep[k] = depth;
		first[x] = k;
		for (nod *p = L[x]; p != NULL; p = p->urm)
			if (!viz[p->x]){
				euler_det(p->x, depth + 1);
				parc[++k] = x;
				dep[k] = depth;

			};
};

int lca(int x, int y)
{
//LCA-ul nodurilor x si y va fi nodul cu nivel minim din secventa (First[x], First[y]) din reprezentarea Euler a arborelui
	int diff, l, sol, sh;
	int a = first[x], b = first[y];
	if(a > b) swap(a, b);
	diff = b - a + 1;
	l = Lg[diff];
	sol = M[l][a];
	sh = diff - (1 << l);
	if(dep[sol] > dep[M[l][a + sh]])
		sol = M[l][a + sh];
	return parc[sol];
}

int main(){
	
	int x, x1, x2;
	
	fin>>n>>m;
	for (int i = 2; i <= n; ++i){
		fin>>x;
		add_nod(i, x);
	};

	euler_det(1, 0);

	//preprocesarea RMQ	
	Lg[1] = 0;
	for (int i = 2; i <= k; i++)
		Lg[i] = Lg[i >> 1] + 1;
	rmq();

	//efectuarea query-urilor
	/*for (int i = 1; i <= k; ++i)
		fout<<parc[i]<<' ';
	fout<<'\n';
	for (int i = 1; i <= k; ++i)
		fout<<first[i]<<' ';
	fout<<'\n';*/
	for (int i = 1; i <= m; ++i){
		fin>>x1>>x2;
		fout<<lca(x1,x2)<<'\n';
	};
	
	return 0;
};