倍增法求最近公共祖先(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;kint 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]int depdif=dep[x]-dep[y];
    for(int i=0;i<="log base 2 of maxn";i++)
    {
        if((1<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;
}

你可能感兴趣的:(高二复习,-,2016,lca,倍增,算法)