P3379 【模板】最近公共祖先(LCA)
非常详细的题解见洛谷,个人见解见代码
方法一:倍增法
#include<bits/stdc++.h>
using namespace std;
#define N 500005
vector<int> G[N];//链树,以链上的元素为根节点的树
void add(int x,int y)
{
G[x].push_back(y);
G[y].push_back(x);
}
int fa[N][21]={0};
int depth[N]={0};
int n,m,s;
void proc()//预处理
{
for(int k=1;k<=18;k++)//直接以最大常数算,反正也快不了多少
for(int i=1;i<=n;i++)
fa[i][k]=fa[fa[i][k-1]][k-1];//如果超过了,父节点就为零,即代表没有父节点
}
void build(int now,int fath)
{
fa[now][0]=fath;
for(int i=0;i<G[now].size();i++)
{
int next=G[now][i];
if(next!=fath)//边不通来路
{
depth[next]=depth[now]+1;
build(next,now);
}
}
}
int lca(int x,int y)
{
if(depth[x]<depth[y])swap(x,y);
for(int i=18;i>=0;i--)
if(depth[fa[x][i]]>=depth[y])//最终得到的一定是相等的
x=fa[x][i];//如果depth为零,代表该节点不存在
//printf("%d,%d\n",x,y);
if(x==y)return x;//如果不加这个就会返回他们的父节点,就不是最近祖先了
//任意两个节点的父节点至少是s,所以fa[s]可以不用管
for(int i=18;i>=0;i--)
if(fa[x][i]!=fa[y][i])
{
x=fa[x][i];
y=fa[y][i];
}//上浮到离lca最近的点
return fa[x][0];
}
int main()
{
scanf("%d%d%d",&n,&m,&s);
for(int i=1;i<=n-1;i++)
{
int x,y;
scanf("%d%d",&x,&y);
add(x,y);
}
depth[s]=1;//要与不存在的节点分开来
build(s,0);
proc();
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",lca(x,y));
}
return 0;
}
方法二:tarjan法(但是蒟蒻只能理解成集合法)
#include<bits/stdc++.h>
using namespace std;
#define N 500005
vector<int> G[N];
vector< pair<int,int> > Q[N];
int ans[N]={0};
int vis[N]={0};
int fa[N]={0};
int finds(int now)
{
return now==fa[now]?now:fa[now]=finds(fa[now]);
}
void ss(int now)
{
fa[now]=now;
vis[now]=1;
for(int i=0;i<G[now].size();i++)
{
int next=G[now][i];
if(!vis[next])
{
ss(next);//lca有两种,一种在一条直链上,另一种不在同一条链上,对于不在同一条链上的情况的解释:ss内出现的且第一次出现的询问都归结到该点
fa[next]=now;
}
}
for(int i=0;i<Q[now].size();i++)
{
int ano=Q[now][i].first;
if(vis[ano])
{
ans[Q[now][i].second]=finds(ano);
}
}
}
void addedge(int x,int y)
{
G[x].push_back(y);
G[y].push_back(x);
}
void addques(int x,int y,int id)
{
Q[x].push_back(make_pair(y,id));
Q[y].push_back(make_pair(x,id));
}
int main()
{
int n,m,s;
scanf("%d%d%d",&n,&m,&s);
for(int i=1;i<=n-1;i++)
{
int x,y;
scanf("%d%d",&x,&y);
addedge(x,y);
}
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
addques(x,y,i);
}
ss(s);
for(int i=1;i<=m;i++)
{
printf("%d\n",ans[i]);
}
return 0;
}

浙公网安备 33010602011771号