Cod sursa(job #379017)

Utilizator Bogdan_tmmTirca Bogdan Bogdan_tmm Data 30 decembrie 2009 12:24:37
Problema Lowest Common Ancestor Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.31 kb
#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
const int nMax=100004;
typedef pair <int,int> p;
int n,m;
vector <int> graph[nMax];
vector <p> a(1);
int lg[nMax+nMax],rmq[19][nMax+nMax],fPoz[nMax];
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);
	for(int i=2;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		graph[x].push_back(i);
	}
	dfs(1,0);
	int N=(int)a.size()-1;
	for(int i=N;i;i--)
	{
		fPoz[a[i].first]=i;
		rmq[0][i]=i;
	}
	for(int i=2;i<N;i++)
	{
		lg[i]=lg[(i>>1)]+1;
	}
	for(int j=1;(1<<j)<=N;j++)
	{
		for(int 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--;)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		int st=min(fPoz[x],fPoz[y]);
		int dr=max(fPoz[x],fPoz[y]);
		p sol=a[rmq[lg[dr-st+1]][st]];
		p 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;
}