Cod sursa(job #1280078)

Utilizator dr_personalityEftime Andrei Horatiu dr_personality Data 1 decembrie 2014 14:11:44
Problema Lowest Common Ancestor Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.48 kb
#include<fstream>
#include<vector>
#include<string>
using namespace std;
ifstream in("lca.in");
ofstream out("lca.out");

const int nmax = 100006;
int n, m, d[18][nmax], nivel[nmax];
vector<int> fii[nmax];
string buffer;
string::iterator buffer_it;

int lca(int x, int y)
{
	if(nivel[x]<nivel[y])
	{
		int p = nivel[y] - nivel[x];

		for(int i = 0; p!=0; p/=2, i++)
            if(p%2==1)
                y = d[i][y];
	}
	else
	{
		int p = nivel[x] - nivel[y];

		for(int i = 0; p!=0; p/=2, i++)
            if(p%2==1)
                x = d[i][x];
	}

	for(int i = 14; i>=0; i--)
	{
		if(d[i][x]!=d[i][y])
		{
			x = d[i][x];
			y = d[i][y];
		}
	}

	if(x==y)
		return x;
	else
		return d[0][x];
}

void dfs(int x, int level)
{
	nivel[x] = level;
	for(int i = 0; i<(int)fii[x].size(); i++)
		dfs(fii[x][i], level + 1);
}


void read_int_nn( int &x ) {
for ( ; *buffer_it>'9' || *buffer_it<'0'; ++buffer_it ) ;
for ( x= 0; *buffer_it>='0' && *buffer_it<='9'; ++buffer_it ) {
x= x*10+*buffer_it-'0';
}
}

int main(){
	int player_unu=0;

	getline(in, buffer, (char)0);
	buffer_it = buffer.begin();

	read_int_nn(n);
	read_int_nn(m);
	for(int i = 2; i<=n; i++)
	{
		read_int_nn(d[0][i]);

		fii[d[0][i]].push_back(i);
	}

	dfs(1, 1);

	for(int i = 1; i<=17; i++)
		for(int j = 1; j<=n; j++)
			d[i][j] = d[i - 1][d[i - 1][j]];

	for(int i = 0; i<m; i++)
	{
		int x, y;
		in>>x>>y;
		out<<lca(x, y)<<'\n';
	}

	return player_unu;
}