P3379 【模板】最近公共祖先（LCA）-（tarjan算法）

题目描述

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

输入格式

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

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

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

输出格式

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

 输入

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

输出

4
4
1
4
4

 

说明/提示

对于 30% 的数据，N≤10，M≤10。

对于 70% 的数据，N≤10000,M≤10000。

对于 100% 的数据，1≤N,M≤500000，1≤x,y,a,b≤N，不保证 a≠b

举个例子，求5和6的最近公共祖先是3，一开始所有的pre[]值都为他们本身，

从1开始，先标记1走过了，然后遍历1的临边，先是2，2没有走过，递归进入并标记2为走过了，再遍历2的临边只有1但是1走过了不进行操作第一个for循环结束，进行第二个for循环遍历2的查询边，发现还是没有，返回到1，并使pre[2]=1,然后继续1的第二个边 3，进入并标记3为走过，再进入3的边4，进入并标记4为走过，进入6并标记为走过，6的情况和2一样，第一个for循环结束，进行第二个for循环遍历6的查询边为5，但5此时没有走过，不做处理，返回到4，并使pre[6]=4,同理pre[7]=4,此时4处理完毕也返回到3，并使pre[4]=3,进入5，注意此时并不是直接开始遍历5 的查询边，而是先处理完5的边8和9，递归返回到5时才开始遍历5的查询边，发现6已经走过了，于是直接find（6）,pre[6]=4,pre[4]=3,pre[3]=3,找祖先得到3，祖先不是1，因为此时的3还没有处理完，还有5没有递归返回，所以3也没有返回，所以pre[3]还是等于它本身为3。

AC代码：

#include<iostream>
#include<vector>
using namespace std;
typedef struct _node{
	int to;
	int index;//下标，第几个查询
}Node;
vector<int> e[500010];//存图
vector<Node> from[500010];//存查询
int pre[500010];//记录父节点
int mark[500010];
int ans[500010];//存答案
int n,m,s;
int find(int x){//找祖先节点
    return pre[x]==x?x:pre[x]=find(pre[x]);
//路径压缩，在洛谷上第一次没有压缩没过，第二次压缩了就过了
}
void tarjan(int u){
	int i;
	int o;
	mark[u]=1;
	for(i=0;i<e[u].size();i++){//遍历临边
		int v=e[u][i];
		if(!mark[v]){//如果没有走过，则递归进入
			tarjan(v);
			pre[v]=u;//递归返回后改变v的pre[]值为父节点u
		}
	}
	for(o=0;o<from[u].size();o++){//遍历查询边
		Node tmp=from[u][o];
		if(mark[tmp.to]==1){
			ans[tmp.index]=find(tmp.to);
		}
	}
}
int main(){
	scanf("%d%d%d",&n,&m,&s);
	int i;
	for(i=0;i<n-1;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		e[y].push_back(x);
		e[x].push_back(y);
	}

   for(i=0;i<m;i++){
   	    int a,b;
   	    scanf("%d%d",&a,&b);
       	from[a].push_back({b,i});
       	from[b].push_back({a,i});
   }

   for(i=1;i<=n;i++){
       	pre[i]=i;
   }

   tarjan(s);

   for(i=0;i<m;i++){
   		printf("%d\n",ans[i]);
   }
	return 0;
}