tarjan求LCA笔记

今天学了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;
}