AIZU Lowest Common Ancestor Tarjan求LCA

最新推荐文章于 2021-06-29 10:43:34 发布

NeighThorn

最新推荐文章于 2021-06-29 10:43:34 发布

阅读量264

点赞数 1

分类专栏： AOJ

本文链接：https://blog.csdn.net/neighthorn/article/details/53129631

版权

AOJ 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

今天刚刚会写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;   
}