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

地址：https://www.luogu.org/problemnew/show/P3379

思路：LCA模板题-需要用链式向前星优化边集

一、离线LCA-Tarjan算法

二、在线LCA-倍增DP

Code 一:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long LL;

const int MAX_N=5e5+5;
const int MAX_Q=5e5+5;
struct node{
	int w;
	int to;
	int next;
};

int n,Q;
int id[MAX_N];
int cnt,cntq;
int head[MAX_N],headq[MAX_N];
node edge[MAX_N<<1],que[MAX_Q<<1];
int res[MAX_Q];
bool boo[MAX_N];

void add(int u,int v,int w){
	edge[cnt].w=w;
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}

void addq(int u,int v,int w){
	que[cntq].w=w;
	que[cntq].to=v;
	que[cntq].next=headq[u];
	headq[u]=cntq++;
}

int Find(int x){
	if(id[x]!=x)	id[x]=Find(id[x]);
	return id[x];
}

void DFS(int u,int pre);
int main()
{
	memset(head,-1,sizeof(head));
	memset(headq,-1,sizeof(headq));
	int u,v,S;
	scanf("%d%d%d",&n,&Q,&S);
	for(int i=1;i<n;++i)
	{
		scanf("%d%d",&u,&v);
		add(u,v,0);
		add(v,u,0);
		id[i]=i;
	}
	id[n]=n;
	for(int i=1;i<=Q;++i)
	{
		scanf("%d%d",&u,&v);
		addq(u,v,i);
		addq(v,u,i);
	}
	DFS(S,0);
	for(int i=1;i<=Q;++i)
		printf("%d\n",res[i]);
	
	return 0;
}

void DFS(int u,int pre)
{
	boo[u]=true;
	for(int i=head[u];i!=-1;i=edge[i].next)
		if(edge[i].to!=pre){
			DFS(edge[i].to,u);
		}
	for(int i=headq[u];i!=-1;i=que[i].next)
		if(boo[que[i].to]){
			res[que[i].w]=Find(que[i].to);
		}
	id[u]=pre;
}

Code 二：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;

const int MAX_N=5e5+5;
const int MAX_M=5e5+5;
struct node{
	int w;
	int to;
	int next;
};
int n,m,S,cnt;
int head[MAX_N];
node edge[MAX_M<<1];
int h[MAX_N];
int dp[MAX_N][20];	//dp[i][j]:节点i的第2^j的祖先节点 

void add(int u,int v){
	edge[cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void DFS(int u,int pre);
int LCA(int u,int v);
int main()
{
	memset(head,-1,sizeof(head));
	scanf("%d%d%d",&n,&m,&S);
	int u,v;
	for(int i=1;i<n;++i)
	{
		scanf("%d%d",&u,&v);
		add(u,v);	add(v,u);
	}
	DFS(S,0);
	for(int j=1;(1<<j)<=n;++j)
		for(int i=1;i<=n;++i) 
			dp[i][j]=dp[dp[i][j-1]][j-1];
	for(int i=0;i<m;++i)
	{
		scanf("%d%d",&u,&v);
		printf("%d\n",LCA(u,v));
	}
	return 0;
}

void DFS(int u,int pre)
{
	int v;
	for(int i=head[u];i!=-1;i=edge[i].next)
		if(edge[i].to!=pre){
			v=edge[i].to;
			h[v]=h[u]+1;
			dp[v][0]=u;
			DFS(v,u);
		}
}

int LCA(int u,int v)
{
	if(h[u]>h[v])	swap(u,v);
	int t=h[v]-h[u],x;
	while(t){	//u,v调整到同一深度 
		x=log2(t);
		v=dp[v][x];
		t=h[v]-h[u];
	}
	if(u!=v){
		for(int i=log2(h[u]);i>=0;--i)
			if(dp[u][i]!=dp[v][i]){
				u=dp[u][i];	v=dp[v][i];
			}
		u=dp[u][0];
	}
	return u;
}