当前位置: 代码迷 >> 综合 >> dfs(tarjan)求lca
  详细解决方案

dfs(tarjan)求lca

热度:56   发布时间:2023-10-29 18:10:29.0

离线做法,可以做到o(n+m)
对于每一棵树,向下搜索,变成小树,如果子节点没有子节点了,就返回
然后在每一步中,找有x的查询,如果另一个访问过,就是他们的祖先
不同颜色的就是不同的子树
这里写图片描述

#include<cstdio>
#include<iostream>
using namespace std;
const int M=1001009;
int n,m,s,f[M],vis[M],lca[M*2];
int nex[M*2],head[M*2],to[M*2],tot,heade[M*2],nexe[M*2],toe[M*2],tote;
void add(int x,int y){nex[++tot]=head[x];to[tot]=y;head[x]=tot;
}
void adde(int x,int y){nexe[++tote]=heade[x];toe[tote]=y;heade[x]=tote;
}
int find(int x){return f[x]==x?x:f[x]=find(f[x]);
}
void dfs(int x){f[x]=x;//小树的根vis[x]=1;for(int i=head[x];i;i=nex[i]){int tmp=to[i];if(!vis[tmp]){dfs(tmp);//建小树f[tmp]=x;//把爸爸改回来}}for(int i=heade[x];i;i=nexe[i]){int tmp=toe[i];if(vis[tmp]){lca[i]=find(tmp);if(i%2) lca[i+1]=lca[i];//两个的祖先相同else lca[i-1]=lca[i];//}}
}
int main(){scanf("%d%d%d",&n,&m,&s);for(int i=1;i<n;i++){int x,y;scanf("%d%d",&x,&y);add(x,y);add(y,x);}for(int i=1;i<=m;i++){int x,y;scanf("%d%d",&x,&y);adde(x,y);adde(y,x);}dfs(s);for(int i=1;i<=m;i++)printf("%d\n",lca[i*2]);}