poj 1330 Nearest Common Ancestors （最简单的LCA）

题意：

给出一棵树的结构。

给出两个点X和Y，求它俩的LCA。

思路：

只需求两个点的LCA，用了两种方法，一种离线tarjan，一种直接搞。

看代码。

代码：

方法一：直接搞。

int const maxn = 10005;

int T,n,a,b;
int fa[maxn];
int X,Y;

int main(){
    cin>>T;
    while(T--){
        scanf("%d",&n);
        mem(fa,-1);
        rep(i,1,n-1){
            scanf("%d%d",&a,&b);
            fa[b]=a;
        }
        scanf("%d%d",&X,&Y);
        map<int,char> mp;
        int t1=X;
        while(t1!=-1){
            mp[t1]=1;
            t1=fa[t1];
        }
        t1=Y;
        while(t1!=-1){
            if(mp[t1]==1){
                printf("%d\n",t1);
                break;
            }
            t1=fa[t1];
        }
    }
}

方法二：离线tarjan。

int const maxn = 10005;

struct node{
    int to,w,next,lca;
};
node edge[maxn*2];

bool vis[maxn];
int cnt1,cnt2;
int head[maxn], fa[maxn];
bool flag;
int X,Y,n;

inline void Addedge(int u,int v,int w){
    edge[++cnt1].w=w;
    edge[cnt1].to=v;
    edge[cnt1].next=head[u];
    head[u]=cnt1;
}

void init(){
    mem(head,-1);
    mem(vis,false);
    cnt1=0;
    flag=false;
    rep(i,1,n) fa[i]=i;
}

int findFa(int x){
    if(fa[x]==x) return fa[x];
    return fa[x]=findFa(fa[x]);
}
void Tarjan_LCA(int u){ //离线LCA算法
    if(flag)
        return;
    fa[u]=u;
    vis[u]=true;
    for(int i=head[u];i!=-1;i=edge[i].next){
        if(!vis[edge[i].to]){
            Tarjan_LCA(edge[i].to);
            fa[edge[i].to]=u;
        }
    }
    if(u==Y && !flag){
        if(vis[X]){
            printf("%d\n",findFa(X));
            flag=true;
            return;
        }
    }
    if(u==X && !flag){
        if(vis[Y]){
            printf("%d\n",findFa(Y));
            flag=true;
            return;
        }
    }
}

int T,a,b;

int main(){
    cin>>T;
    while(T--){
        scanf("%d",&n);
        init();
        bool t1[maxn]={0};
        rep(i,1,n-1){
            scanf("%d%d",&a,&b);
            Addedge(a,b,1);
            t1[b]=true;
        }
        scanf("%d%d",&X,&Y);
        rep(i,1,n) if(!t1[i]){
            Tarjan_LCA(i); //i为树的根结点
            break;
        }
    }
}