Tarjan求LCA

LCA

树上最近公共祖先的简称。可以是父节点，也可以是自己。

算法思路

这种求LCA的算法比较常用，而且是离线的。需要用到并查集数据类型。算法基本流程为
1.访问点u。
2.标记访问，并认自己为爹。
3.遍历点u的所有儿子。如果儿子还有儿子，则再访问儿子的所有儿子。
4.查询所有与该点有关的询问。若另一个点已被访问，找到对方的爹，就是lca。
由此可以看出此算法是离线的，可以一次性处理完所有询问，所以其时间复杂度也比较小。

相关复杂度

时间复杂度为o(m+n)

代码

#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,q,tol,tolq,p;
int fa[5005];
struct edge{
 int nxt;
 int to;
 int lca;
}e[5005];//结构体存边 
struct edge qs[5005];
int hd[5005],hdq[5005];
bool vis[5005];
void add(int u,int v)//u to v,end at 0
{
 tol++;
 e[tol].nxt=hd[u];
 e[tol].to=v;
 hd[u]=tol;
}
void addq(int u,int v)//u to v,end at 0
{
 tolq++;
 qs[tolq].nxt=hdq[u];
 qs[tolq].to=v;
 hdq[u]=tolq;
}
int getfa(int t)//找爹函数
{
 return fa[t]==t?t:getfa(fa[t]);
} 
void tarjan(int u)
{
 vis[u]=1;
 fa[u]=u;//注意顺序 
 for(int i=hd[u];i;i=e[i].nxt)
 {
  int v=e[i].to;
  if(!vis[v])
  {
   tarjan(v);
   fa[v]=u;
  }
 }
 for(int i=hdq[u];i;i=qs[i].nxt)
 {
  int v=qs[i].to;
  if(!qs[i].lca&&vis[v])
  {
   qs[i].lca=getfa(v);
   if(i%2) qs[i+1].lca=qs[i].lca;
   else qs[i-1].lca=qs[i].lca; 
  }
 }
}
int main()
{
 scanf("%d%d%d",&n,&m,&p);//需输入根节点 
 for(int i=1;i<=m;i++)
 {
  int x,y;
  scanf("%d%d",&x,&y);
  add(x,y);
  add(y,x);
 }
 scanf("%d",&q);
 for(int i=1;i<=q;i++)
 {
  int x,y;
  scanf("%d%d",&x,&y);
  addq(x,y);
  addq(y,x);
 }
 tarjan(p);
 for(int i=1;i<=q;i++)
 printf("%d\n",qs[i*2].lca);
 return 0;
}