AIZU Lowest Common Ancestor Tarjan求LCA

最新推荐文章于 2018-08-29 16:21:59 发布

原创最新推荐文章于 2018-08-29 16:21:59 发布 · 297 阅读

0 ·

CC 4.0 BY-SA版权

AOJ 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个使用Tarjan算法实现的离线求最近公共祖先(LCA)的问题解决方案。该实现首先进行预处理建立图结构，然后通过Tarjan算法进行遍历，最终输出所有查询的答案。

今天刚刚会写Tarjan离线LCA…(捂脸…)

代码如下：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
//by NeighThorn
using namespace std;
const int maxn=100000+5;
int n,q,hd[maxn],to[maxn*2],nxt[maxn*2],cnt,vis[maxn],fa[maxn],ans[maxn];
struct M{
    int y,id;
    M(int a=0,int b=0){
        y=a,id=b;
    }
};
vector<M> v[maxn];
inline int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);    
}
inline void add(int x,int y){
    to[cnt]=y;nxt[cnt]=hd[x];hd[x]=cnt++;   
}
inline void tarjan(int root){
    vis[root]=1;
    for(int i=0;i<(int)v[root].size();i++)
        if(vis[v[root][i].y])
            ans[v[root][i].id]=find(v[root][i].y);
    for(int i=hd[root];i!=-1;i=nxt[i])
        if(!vis[to[i]])
            tarjan(to[i]),fa[to[i]]=root;
}
signed main(void){
    scanf("%d",&n);cnt=0;
    memset(hd,-1,sizeof(hd));
    for(int i=0,k;i<n;i++){
        scanf("%d",&k);
        for(int j=1,x;j<=k;j++)
            scanf("%d",&x),add(i,x),add(x,i);
    }scanf("%d",&q);
    for(int i=0;i<n;i++)
        fa[i]=i,vis[i]=0;
    for(int i=1,x,y;i<=q;i++)
        scanf("%d%d",&x,&y),v[x].push_back(M(y,i)),v[y].push_back(M(x,i));
    tarjan(0);
    for(int i=1;i<=q;i++)
        printf("%d\n",ans[i]);
    return 0;   
}