POJ1330 TarjanLCA

最新推荐文章于 2019-11-29 01:10:50 发布

原创最新推荐文章于 2019-11-29 01:10:50 发布 · 414 阅读

0 ·

CC 4.0 BY-SA版权

LCA 专栏收录该内容

1 篇文章

订阅专栏

本文分享了一种离线最低公共祖先(LCA)算法的实现思路，并提供了完整的C++代码示例。通过构造树结构并利用Tarjan算法，有效地解决了LCA问题。

LCA裸题。
试着写了一下离线LCA算法。
自己的代码：

#include<cstdio>
#include<cstring>
using namespace std;
int n,t;
bool rt[10001];
struct e
{
    int t;
    e *n;
    e(int t,e *n):t(t),n(n){}
}*f[10001]={};
namespace memo
{
    e *stk[10001];
    int top=-1;
    void recycle(int n)
    {
        for(int i=1;i<=n;i++)
            if(f[i])
            {
                stk[++top]=f[i];
                f[i]=0;
            }
    }
    e *alloc(int t,e *n)
    {
        if(top!=-1)
        {
            e *prov=stk[top];
            stk[top]=stk[top]->n;
            if(!stk[top]) --top;
            prov->t=t;prov->n=n;
            return prov;
        }
        return new e(t,n);
    }
}
namespace tarjan
{
    int ans;
    int r[10001];
    bool vis[10001];
    int aim1,aim2;
    int find(int x)
    {return r[r[x]]==r[x]?r[x]:r[x]=find(r[x]);}
    inline void link(int a,int b)
    {
        r[find(b)]=find(a);
    }
    void dfs(int x)
    {
        vis[x]=1;
        for(e* i=f[x];i;i=i->n)
        {
            dfs(i->t);
            link(x,i->t);
        }
        if(x==aim1&&vis[aim2]) ans=find(aim2);
        else if(x==aim2&&vis[aim1]) ans=find(aim1);
    }
    void work(int n,int a,int b)
    {
        memset(vis,0,n+1);
        aim1=a;aim2=b;
        int root;
        for(int i=1;i<=n;i++)
        {
            r[i]=i;
            if(rt[i]) root=i;
        }
        dfs(root);
    }
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        int a,b;
        memset(rt,1,n+1);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            rt[b]=0;
            f[a]=memo::alloc(b,f[a]);
        }
        scanf("%d%d",&a,&b);
        for(int i=1;i<=n;i++)
        {
            if(rt[i]) {tarjan::work(n,a,b);break;}
        }
        printf("%d\n",tarjan::ans);
        memo::recycle(n);
    }
    return 0;
}