RQNOJ 28 LCA问题

查看题目 Show Problem

[Donate]您的捐助，让RQNOJ明天更美好!

题目:[Stupid]愚蠢的宠物

问题编号:28

题目描述

背景
大家都知道，sheep有两只可爱的宠物（一只叫神牛，一只叫神菜）。有一天，sheep带着两只宠物到狗狗家时，这两只可爱的宠物竟然迷路了……

描述
狗狗的家因为常常遭到猫猫的攻击，所以不得不把家里前院的路修得非常复杂。狗狗家前院有N个连通的分叉结点，且只有N-1条路连接这N个节点，节点的编号是1-N（1为根节点）。sheep的宠物非常笨，他们只会向前走，不会退后（只向双亲节点走），sheep想知道他们最早什么时候会相遇（即步数最少）。

N的范围《=1000000

输入格式

第1行：一个正整数N，表示节点个数。
第2~N行：两个非负整数A和B，表示A是B的双亲。（保证A,B<=n）
第N+1行：两个非负整数A和B，表示两只宠物所在节点的位置。（保证A,B<=n）

输出格式

输出他们最早相遇的节点号。

样例输入

10 1 2 1 3 1 4 2 5 2 6 3 7 4 8 4 9 4 10 3 6

样例输出

1

这个题是非常裸的LCA问题。我本来想用RMQ来写的

但是突然发现询问只有一次。。FT。。

所以其实暴力的O（n）枚举是最快的啦。。因为RMQ还要初始化，初始化的时间都比这个长了。

我的代码：

#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
#define maxn 1000005

using namespace std;

vector<int>map[maxn];
int pos[maxn];
int deep[2*maxn];
int eular[2*maxn];
bool used[maxn];
int n,num;

void init()
{
	int i;
	for(i=0;i<=n;i++)
	{
		map[i].clear();
		pos[i]=-1;
		used[i]=false;
	}
	num=0;
}

void dfs(int root,int dep)
{
	int i;
	used[root]=true;
	eular[num]=root;
	deep[num]=dep;
	num++;
	for(i=0;i<map[root].size();i++)
	{
		if(!used[map[root][i]])
		{
			dfs(map[root][i],dep+1);
			eular[num]=root;
			deep[num]=dep;
			num++;
		}
	}
}

int main()
{
	int i,a,b;
	while(scanf("%d",&n)!=EOF)
	{
		init();
		for(i=1;i<=n-1;i++)
		{
			scanf("%d%d",&a,&b);
			map[a].push_back(b);
			map[b].push_back(a);
		}
		dfs(1,0);
		scanf("%d%d",&a,&b);
		for(i=0;i<num;i++)
		{
			if(pos[eular[i]]==-1)
				pos[eular[i]]=i;
		}
		if(pos[a]>pos[b])
			swap(a,b);
		int min=99999999,ans;
		for(i=pos[a];i<=pos[b];i++)
			if(deep[i]<min)
			{
				min=deep[i];
				ans=eular[i];
			}
		printf("%d\n",ans);
	}
	return 0;
}