倍增法求最近公共祖先(LCA)的算法模板

“log base 2 of maxn”:以2为底maxn的对数.
定义数组:

int bzfa[maxn]["log base 2 of maxn"];

输入:

for(int i=1;i<=n;i++)
{
    int a,b;
    scanf("%d%d",&a,&b);
    bzfa[b][0]=a;
    if(bzfa[a][0]==0)
    {
        root=a;
    }
}

BFS计算dep:

void BFS(int i)
{
    queue<int>q;
    q.push(i);
    vis[i]=1;
    while(!q.empty())
    {
        int t=q.front(); q.pop();
        for(int k=0;k<g[t].size();k++)
        {
            int j=g[t][k];
            if(vis[j]) continue;
            dep[j]=dep[t]+1;
            vis[j]=1;
            q.push(j);
        }
    }
}

预处理bzfa数组:

void BZ_LCA()
{
    for(int j=1;j<="log base 2 of maxn";j++)
    {
        for(int i=1;i<=n;i++)
        {
            bzfa[i][j]=bzfa[bzfa[i][j-1]][j-1];
        }
    }
}

LCA:

int LCA(int x,int y)
{
    if(dep[x]<dep[y])
    {
        swap(x,y);
    }
    int depdif=dep[x]-dep[y];
    for(int i=0;i<="log base 2 of maxn";i++)
    {
        if((1<<i) & depdif)
        {
            x=bzfa[x][i];
        }
    }
    if(x==y) return x;
    for(int i="log base 2 of maxn";i>=0;i--)
    {
        if(bzfa[x][i]!=bzfa[y][i])
        {
            x=bzfa[x][i];
            y=bzfa[y][i];
        }
    }
    x=bzfa[x][0];
    return x;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值