Cod sursa(job #379025)

Utilizator Bogdan_tmmTirca Bogdan Bogdan_tmm Data 30 decembrie 2009 12:48:39
Problema Lowest Common Ancestor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.3 kb
#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
const int nMax=100004;
typedef pair <int,int> p;
int n,m;
p sol,sol2;
vector <int> graph[nMax];
vector <p> a(1);
int rmq[18][(nMax<<1)];
void dfs(int nod,int gr)
{
	a.push_back(p(nod,gr));
	if(graph[nod].empty())
	{
		return;
	}
	else
	{
		for(vector <int> ::iterator it=graph[nod].begin();it!=graph[nod].end();it++)
		{
			dfs(*it,gr+1);
			a.push_back(p(nod,gr));
		}
	}
}
int main()
{
	freopen("lca.in","r",stdin);
	freopen("lca.out","w",stdout);
	scanf("%d%d",&n,&m);
	int lg[(nMax<<1)],fPoz[nMax];
	int i,j,x,y,st,dr;
	for(i=2;i<=n;i++)
	{
		scanf("%d",&x);
		graph[x].push_back(i);
	}
	dfs(1,0);
	int N=(int)a.size()-1;
	for(i=N;i;i--)
	{
		fPoz[a[i].first]=i;
		rmq[0][i]=i;
	}
	for(i=2;i<=N;i++)
	{
		lg[i]=lg[(i>>1)]+1;
	}
	for(j=1;(1<<j)<N;j++)
	{
		for(i=1;i<=N-(1<<j)+1;i++)
		{
			rmq[j][i]=rmq[j-1][i];
			if(a[rmq[j-1][i+(1<<(j-1))]].second<a[rmq[j][i]].second)
				rmq[j][i]=rmq[j-1][i+(1<<(j-1))];
		}
	}
	for(;m--;)
	{
		scanf("%d%d",&x,&y);
		st=fPoz[x];
		dr=fPoz[y];
		if(st>dr)
			swap(st,dr);
		sol=a[rmq[lg[dr-st+1]][st]];
		sol2=a[rmq[lg[dr-st+1]][dr-(1<<lg[dr-st+1])+1]];
		if(sol2.second<sol.second)
			sol=sol2;
		printf("%d\n",sol.first);
	}
	return 0;
}