Lowest Common Ancestor SPOJ - LCA LCA 树上倍增

一道模板题

讲解请看 https://blog.youkuaiyun.com/qq_41837216/article/details/83416270

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=1010;
vector<int>g[maxn];
int fa[maxn][11],d[maxn];
int n,m,root;
bool vis[maxn];
void init()
{
    memset(fa,0,sizeof(fa));
    memset(d,0,sizeof(d));
    memset(vis,0,sizeof(vis));
}
void dfs(int u)
{
    int i;
    vis[u]=1;
    for(i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(!vis[v])
        {
            d[v]=d[u]+1;
            dfs(v);
        }
    }
}
void bz()
{
    int i,j;
    for(j=1;j<=10;j++)
        for(int i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1];
}
int lca(int u,int v)
{
    if(d[u]<d[v])
        swap(u,v);
    int dc=d[u]-d[v];
    int i;
    for(i=0;i<=10;i++)
    {
        if((1<<i)&dc)
            u=fa[u][i];
    }
    if(u==v)
        return u;
    for(i=10;i>=0;i--)
    {
        if(fa[u][i]!=fa[v][i])
        {
            u=fa[u][i];
            v=fa[v][i];
        }
    }
    u=fa[u][0];
    return u;
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int i=1;i<=t;i++)
    {
        init();
        scanf("%d",&n);
        for(int j=0;j<=n;j++)
            g[j].clear();
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&m);
            while(m--)
            {
                int a;
                scanf("%d",&a);
                g[j].push_back(a);
                fa[a][0]=j;
                if(fa[j][0]==0)
                    root=j;
            }
        }
        d[root]=1;
        dfs(root);
        bz();
        int k,x,y;
        scanf("%d",&k);
        printf("Case %d:\n",i);
        while(k--)
        {
            scanf("%d%d",&x,&y);
            printf("%d\n",lca(x,y));
        }

    }
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值