Cod sursa(job #393614)

Utilizator Alexa_ioana_14Antoche Ioana Alexandra Alexa_ioana_14 Data 9 februarie 2010 18:53:14
Problema Lowest Common Ancestor Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.55 kb
#include<cstdio>
#include<vector>
#define pb push_back
using namespace std;
#define minn(a,b) ((a<b)?a:b)
#define N 100001
int e[19][N<<1],n,m,x,y,dep[19][N<<1],lung,jum,pr[N],log[N<<1];
int euler[N<<1],nr;
vector<int>arb[N];

void df(int x,int d)
{
	euler[++nr] = x;
	dep[0][nr]=d;
	if (!pr[x])
		pr[x]=nr;
	size_t g=arb[x].size();
	for (size_t i=0; i<g; ++i)
	{
		df(arb[x][i],1+d);
		euler[++nr] = x;
		dep[0][nr]=d;
	}
}
void rmq()
{
	int i;
	log[2]=1;
	for (i=1;i<=nr;++i)
		e[0][i] = euler[i];
	for(i=3; i<=nr; ++i)
		log[i]=log[i>>1]+1;
	for (int i=1; i<=log[nr]; ++i)
	{
		lung=1<<i;
		jum=lung>>1;
		for(int j=1; j+lung-1<=nr; ++j)
			if(dep[i-1][j]<dep[i-1][j+jum])
			{
				dep[i][j]=dep[i-1][j];
				e[i][j]=e[i-1][j];
			}
			else
			{
				dep[i][j]=dep[i-1][j+jum];
				e[i][j]=e[i-1][j+jum];
			}
	}
}
void solve()
{
	if (pr[x]>pr[y])
		swap(x,y);
	int dim=1<<log[pr[y]-pr[x]+1];
	int lo=log[pr[y]-pr[x]+1];
	if(dep[lo][pr[x]]<dep[lo][pr[y]-dim+1])
		x=e[lo][pr[x]];
	else
		x=e[lo][pr[y]-dim+1];
}
void citire()
{
	freopen("lca.in","r",stdin);
	freopen("lca.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=2; i<=n; ++i)
	{
		scanf("%d",&x);
		arb[x].pb(i);
	}
	df(1,0);
	/*for (int i=1; i<=nr; ++i)
		printf("%d %d\n",e[0][i],dep[0][i]);*/
	rmq();
	/*for (int i=0; i<=log[nr]; ++i)
	{
		for(int j=1; j<=nr; ++j)
			printf("%d ",dep[i][j]);
		printf("\n");
	}*/
	while (m--)
	{
		scanf("%d%d",&x,&y);
		solve();
		printf("%d\n",x);
	}
}
int main()
{
	citire();
	return 0;
}