P3379 【模板】最近公共祖先（LCA）

P3379 【模板】最近公共祖先（LCA）

题目描述

如题，给定一棵有根多叉树，请求出指定两个点直接最近的公共祖先。

输入格式

第一行包含三个正整数N、M、S，分别表示树的结点个数、询问的个数和树根结点的序号。

接下来N-1行每行包含两个正整数x、y，表示x结点和y结点之间有一条直接连接的边（数据保证可以构成树）。

接下来M行每行包含两个正整数a、b，表示询问a结点和b结点的最近公共祖先。

输出格式

输出包含M行，每行包含一个正整数，依次为每一个询问的结果。

输入输出样例

输入 #1复制

5 5 4
3 1
2 4
5 1
1 4
2 4
3 2
3 5
1 2
4 5

输出 #1复制

4
4
1
4
4

说明/提示

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=10，M<=10

对于70%的数据：N<=10000，M<=10000

对于100%的数据：N<=500000，M<=500000

样例说明：

该树结构如下：

第一次询问：2、4的最近公共祖先，故为4。

第二次询问：3、2的最近公共祖先，故为4。

第三次询问：3、5的最近公共祖先，故为1。

第四次询问：1、2的最近公共祖先，故为4。

第五次询问：4、5的最近公共祖先，故为4。

故输出依次为4、4、1、4、4。

LCA模板：

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e6+5;
int numadd=0;
int querynum=0;
int pre[maxn];
bool vis[maxn];
int head[maxn];
int ahead[maxn];
int queryhead[maxn];
int find(int x)
{
	return x==pre[x]?x:pre[x]=find(pre[x]);
}
struct node
{
	int to;
	int from;
	int next;
	int LCA;
} sz[maxn],quesz[maxn];
void add(int from,int to)
{
	sz[++numadd].next=head[from];
	sz[numadd].to=to;
	head[from]=numadd;
}
void add1(int from,int to)
{
	quesz[++querynum].next=ahead[from];
	quesz[querynum].to=to;
	ahead[from]=querynum;
}
void dfs(int x)
{
	pre[x]=x;//初始化 
	vis[x]=1;//已经走过 
	for(int i=head[x]; i; i=sz[i].next)
	{
		if(vis[sz[i].to]==0)
		{
			dfs(sz[i].to);//如果没走过 ，就往下走 
			pre[sz[i].to]=x;//把能走到的点的pre改为x 
		}
	}
	for(int i=ahead[x];i;i=quesz[i].next)
	{
		if(vis[quesz[i].to]==1)
		{
			quesz[i].LCA=find(quesz[i].to);//如果已经被走过直接遍历并查集找到相等的pre[x]=x 
			if(i&1)
			{
				quesz[i+1].LCA=quesz[i].LCA;
			}
			else
			{
				quesz[i-1].LCA=quesz[i].LCA;
			}
		}
	}
}
int main()
{
	int n,m,q;
	scanf("%d %d %d",&n,&m,&q);
	for(int i=0; i<n-1; i++)
	{
		int x,y;
		scanf("%d %d",&x,&y);
		add(x,y);
		add(y,x);
	}
	for(int i=0; i<m; i++)
	{
		int x,y;
		scanf("%d %d",&x,&y);
		add1(x,y);
		add1(y,x);
	}
	dfs(q);
	for(int i=1;i<=m;i++)//对于一次查询x,y。x,y的答案是一样的，所以输出一次就可以了 
	{
		cout<<quesz[i*2].LCA<<endl;
	}
	return 0;
}