POJ 1330 Nearest Common Ancestors(LCA Tarjan)

Nearest Common Ancestors

题意：求两个节点的最近公共父节点

思路：LCA tarjan

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 10010
struct Edge{
    int v;
    int len;
    int next;
}edge[N*2];
int e_tot,e_head[N],root[N];
void e_add(int a,int b,int c){
    edge[e_tot].v=b;
    edge[e_tot].len=c;
    edge[e_tot].next=e_head[a];
    e_head[a]=e_tot++;

    edge[e_tot].v=a;
    edge[e_tot].len=c;
    edge[e_tot].next=e_head[b];
    e_head[b]=e_tot++;

}
struct Query{
    int v;
    int id;
    int next;
}que[N*2];
int q_tot,q_head[N];
void q_add(int a,int b,int c){
    que[q_tot].v=b;
    que[q_tot].id=c;
    que[q_tot].next=q_head[a];
    q_head[a]=q_tot++;

    que[q_tot].v=a;
    que[q_tot].id=c;
    que[q_tot].next=q_head[b];
    q_head[b]=q_tot++;
}
bool vis[N];
int pre[N],ans[N],n;
void init(){
    e_tot=q_tot=0;
    memset(vis,false,sizeof(vis));
    memset(root,true,sizeof(root));
    memset(e_head,-1,sizeof(e_head));
    memset(q_head,-1,sizeof(q_head));
    memset(ans,-1,sizeof(ans));
    for(int i=1;i<=n;i++)
        pre[i]=i;
}
int find(int x){
    return pre[x]==x?pre[x]:pre[x]=find(pre[x]);
}
void Union(int x,int y){
    x=find(x);
    y=find(y);
    if(x!=y){
        pre[y]=x;
    }
}
void Tarjan(int cur){
    vis[cur]=true;
    for(int i=e_head[cur];i!=-1;i=edge[i].next){
        int tmp = edge[i].v;
        if(!vis[tmp]){
            Tarjan(tmp);
            Union(cur,tmp);
        }
    }
    for(int i=q_head[cur];i!=-1;i=que[i].next){
        int tmp = que[i].v;
        if(vis[tmp]){
            ans[que[i].id]=find(tmp);
        }
    }
}
int main(){
    int T,a,b,c,q,m;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        init();
        for(int i=0;i<n-1;i++){
            scanf("%d%d",&a,&b);
            e_add(a,b,0);
            root[b]=false;
        }

        q=1;
        for(int i=0;i<q;i++){
            scanf("%d%d",&a,&b);
            q_add(a,b,i);
        }
        for(int i=1;i<=n;i++){
            //找到根节点后求解
            if(!vis[i]&&root[i]) Tarjan(i);
        }
        for(int i=0;i<q;i++){
            printf("%d\n",ans[i]);
        }
    }
    return 0;
}