洛谷 P3379 【模板】最近公共祖先（LCA）( tarjan算法, 模板题)

最近公共祖先LCA, tarjan算法
本题要点：
1、tarjan算法 是离线算法，把所有的查询读入，最后统一输出。
2、数组 vis[k] 表示 k点的访问状态，
vis[k] == 0, 表示未访问
vis[k] == 1, 表示已经访问， 但是未回溯
vis[k] == 2, 表示已经回溯
3、 如果 某次查询 (x, y), 此时点x 正处于 vis[x] == 1 的状态，而y处于 vis[y] == 2 的状态，
那么 lca(x, y) 就是 y点一直往上走，直达遇到第一个祖先 z, vis[z] == 1.
4、 用 fa[x] 来表示 ，x点是从哪个点访问到的。 可以利用并查集的方法，查询到某一点的 第一个
vis 值为1 的祖先节点。 这里需要并查路径压缩。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int MaxN = 500010, MaxM = 2e6 + 10;
int head[MaxN], ver[MaxM], Next[MaxM];
int fa[MaxN];
int vis[MaxN], ans[MaxM], d[MaxN];	// d[x] 表示点x 与根节点root 之间的距离
int n, m, root, tot;
vector<int> query[MaxN], query_id[MaxN];

void add(int x, int y)
{
	ver[++tot] = y, Next[tot] = head[x], head[x] = tot;
}

void add_query(int x, int y, int id)
{
	query[x].push_back(y), query_id[x].push_back(id);
	query[y].push_back(x), query_id[y].push_back(id);
}

int get(int x)
{
	if(fa[x] == x)
		return x;
	return fa[x] = get(fa[x]);
}

void tarjan(int x)
{
	vis[x] = 1;
	for(int i = head[x]; i; i = Next[i])
	{
		int y = ver[i];
		if(vis[y])	continue;
		d[y] = d[x] + 1;
		tarjan(y);
		fa[y] = x;
	}
	int len = query[x].size();
	for(int i = 0; i < len; ++i)
	{
		int y = query[x][i], id = query_id[x][i];
		if(vis[y] == 2)
		{
			ans[id] = get(y); 	
		}
	}
	vis[x] = 2;
}

int main()
{
	int x, y;
	scanf("%d%d%d", &n, &m, &root);
	for(int i = 1; i <= n; ++i)
	{
		fa[i] = i;
	}
	for(int i = 1; i < n; ++i)
	{
		scanf("%d%d", &x, &y);
		add(x, y);
		add(y, x);
	}
	for(int i = 1; i <= m; ++i)
	{
		scanf("%d%d", &x, &y);
		if(x == y)
		{
			ans[i] = x;
		}else{
			add_query(x, y, i);
		}
	}
	tarjan(root);
	for(int i = 1; i <= m; ++i)
		printf("%d\n", ans[i]);
	return 0;
}

/*
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
*/