今天学了tarjan求LCA。。啊,调了一小时发现两个数组名太像打混了。
思路:从根开始DFS,搜到最深处,去查询有没有询问节点。如果有,且那个节点访问过,那么find(那个节点)就是他们的LCA。如果没访问过,则不操作,出栈。合并他和他的父亲节点。
伪代码(来源:http://www.cnblogs.com/JVxie/p/4854719.html)
1 Tarjan(u)//marge和find为并查集合并函数和查找函数
2 {
3 for each(u,v) //访问所有u子节点v
4 {
5 Tarjan(v); //继续往下遍历
6 marge(u,v); //合并v到u上
7 标记v被访问过;
8 }
9 for each(u,e) //访问所有和u有询问关系的e
10 {
11 如果e被访问过;
12 u,e的最近公共祖先为find(e);
13 }
14 }
代码:
#include<bits/stdc++.h>
using namespace std;
const int MAXN=1000005;
int head[MAXN],fa[MAXN],vis[MAXN],head2[MAXN],cnt=0,gg=0,ans[MAXN],rk[MAXN];
struct edge{
int to,next,num;
}e[MAXN],e1[MAXN];
inline void add(int u,int v){e[++cnt]=(edge){v,head[u]},head[u]=cnt;}
inline void add1(int u,int v,int w){e1[++gg]=(edge){v,head2[u],w},head2[u]=gg;}
int find(int x){
return (x==fa[x])?x:fa[x]=find(fa[x]);
}
void makeuu(int n){
for(int i=0;i<=n;i++)fa[i]=i;
}
void unset(int x,int y){
if((x=find(x))==(y=find(y)))return;
fa[y]=x;
}
void dfsl(int u){
vis[u]=1;
for(int i=head[u];i;i=e[i].next){
if(vis[e[i].to])continue;
dfsl(e[i].to);
unset(u,e[i].to);
}
for(int i=head2[u];i;i=e1[i].next){
if(vis[e1[i].to])ans[e1[i].num]=find(e1[i].to);
}
}
int n,m,s,tem,tem1;
int main(){
scanf("%d%d%d",&n,&m,&s);
makeuu(n);
memset(vis,0,sizeof(vis));
for(int i=1;i<=n-1;i++){
scanf("%d%d",&tem,&tem1);
add(tem,tem1);
add(tem1,tem);
}
for(int i=1;i<=m;i++){
scanf("%d%d",&tem,&tem1);
add1(tem,tem1,i);
add1(tem1,tem,i);
}
dfsl(s);
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
return 0;
}