【Luogu P3379】LCA问题的倍增解法

Luogu P3379
题意:对于两个节点,寻找他们的最近公共祖先。
一个显而易见的解法是对于每一个节点我们都往上遍历一遍,记录下它每一个祖先,然后再从另一个节点出发,一步一步往上走,找到以前记录过第一个节点就是这两个节点的LCA
事实上在这样的数据规模下,这种解法的时间复杂度是让人无法接受的。
很容易发现,这样的解法慢在两个节点是一步一步往上走的,也许可以想个办法一次跳一大步?
于是我们引入了倍增法求LCA。
我们知道,对于任意一个整数数,都可以使用2^k^+2^k-1^ +...+2^0^ 来表示,这是不需要证明的,因为二进制能表示十进制的每一个数。
那么我们就可以每一次跳2^k步来优化这个算法。
还有一个问题就是:我们应该从大的开始跳还是从小的开始跳呢?
答案是从大的开始,因为如果从小的开始,就有可能出现需要回溯的情况。
举个例子,如果从小的开始,我们就会凑出 5=1+2+...,发现不对,然后回溯,凑出5=1+4,但是如果从大的开始就可以直接凑5=4+1,只要判断一下加了这个数比要求的数要大就可以不要这个,取更小的。

但是我们会发现在做这件事之前,我们必须要知道两个节点的深度以便跳步,并且还需要预处理每一个节点的2^k^ 级的祖先。
一个DFS就可以轻松完成这个任务。

void add(int sta,int to)
{
    //链式前向星存树
    edge[++cnt].to=to;
    edge[cnt].next=head[sta];
    head[sta]=cnt;
    //head[i]表示以sta为起点的最后一条边的编号 
    //edge[i].next表示与它起点相同的上一条边的编号 
}//如果没有看懂可以寻找其他博主的资料
void dfs(int fa,int now)
{
    deepth[now]=deepth[fa]+1;//记录深度 
    ant[now][0]=fa;//ant[now][i]表示now节点的2^i级祖先 
    for (int i=1;(1<<i)<=deepth[now];i++)
        ant[now][i]=ant[ant[now][i-1]][i-1];
    //关键的一步:now的2^i级祖先就是now的2^(i-1)级祖先的2^(i-1)级祖先
    //理由如下:2^(i-1)+2^(i-1)=2*2^(i-1)=2^i 
    for (int i=head[now];i!=0;i=edge[i].next)
        if (edge[i].to!=fa) dfs(now,edge[i].to); 
        //判断用于防止往回走。 
}

预处理完之后我们就可以开始寻找LCA了,为了方便,我们采用的方法是令x,y节点先跳到同一深度,再统一往上跳。

int lca(int x,int y)
{
    if (deepth[x]<deepth[y])
        swap(x,y);//方便操作,令x为深度较大的那一个 
    while (deepth[x]>deepth[y])
        x=ant[x][lg[deepth[x]-deepth[y]]-1];
    //这样最后一定会跳到同一深度,想一想为什么。 
    //lg[i]数组表示log(2,i)+1 
    if (x==y) return x; 
    for (int i=lg[deepth[x]];i>=0;i--)//开始往上跳 
        if (ant[x][i]!=ant[y][i]) x=ant[x][i],y=ant[y][i];
    //值得注意的时,我们并不能直接跳到他们的公共祖先上,否则可能会导致误判。
    //如果直接往上跳到公共祖先,可能会多跳了。 
    return ant[x][0];
}

最后是完整代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct data
{
    int to,next;
}edge[500005<<1];
int cnt,head[500005],deepth[500005],ant[500005][21],lg[500005],n,m,s,x,y; 
void add(int sta,int to)
{
    //链式前向星存树
    edge[++cnt].to=to;
    edge[cnt].next=head[sta];
    head[sta]=cnt;
    //head[i]表示以sta为起点的最后一条边的编号 
    //edge[i].next表示与它起点相同的上一条边的编号 
}
void dfs(int fa,int now)
{
    deepth[now]=deepth[fa]+1;//记录深度 
    ant[now][0]=fa;//ant[now][i]表示now节点的2^i级祖先 
    for (int i=1;(1<<i)<=deepth[now];i++)
        ant[now][i]=ant[ant[now][i-1]][i-1];
    //关键的一步:now的2^i级祖先就是now的2^(i-1)级祖先的2^(i-1)级祖先
    //理由如下:2^(i-1)+2^(i-1)=2*2^(i-1)=2^i 
    for (int i=head[now];i!=0;i=edge[i].next)
        if (edge[i].to!=fa) dfs(now,edge[i].to); 
        //判断用于防止往回走。 
}
int lca(int x,int y)
{
    if (deepth[x]<deepth[y])
        swap(x,y);//方便操作,令x为深度较大的那一个 
    while (deepth[x]>deepth[y])
        x=ant[x][lg[deepth[x]-deepth[y]]-1];
    //这样最后一定会跳到同一深度,想一想为什么。 
    //lg[i]数组表示log(2,i)+1 
    if (x==y) return x; 
    for (int i=lg[deepth[x]];i>=0;i--)//开始往上跳 
        if (ant[x][i]!=ant[y][i]) x=ant[x][i],y=ant[y][i];
    //值得注意的时,我们并不能直接跳到他们的公共祖先上,否则可能会导致误判。
    //如果直接往上跳到公共祖先,可能会多跳了。 
    return ant[x][0];
}
int main()
{
    scanf("%d%d%d",&n,&m,&s);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);//暂时不确定父子关系,所以要存两条边。 
    }
    dfs(0,s);
    for (int i=1;i<=n;i++) lg[i]=lg[i-1]+(1<<lg[i-1]==i);
    //用于求出log(2,i)+1的近似值,作常数优化 
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        printf("%d\n",lca(x,y));
    }
    return 0;
} 

猜你喜欢

转载自www.cnblogs.com/notscience/p/11796121.html