倍增法求LCA

本文详细介绍了洛谷模板在解决最近公共祖先(LCA)问题时的常数优化和O(nlgn)预处理步骤。通过位运算和动态规划计算lg[i],并利用fa[cur][i]存储节点的2^i级祖先。在LCA查找过程中,通过逐位比较找到公共祖先。博客提供了完整的C++代码实现。
摘要由CSDN通过智能技术生成

在这里插入图片描述

写这篇博客的时候,这个洛谷的模板还没完全过掉,但是嗯…里面有一些想法很有意思,所以写下来记录一下
原题参考:P3379 【模板】最近公共祖先(LCA)
本文涉及思路参考:神犇的题解

首先是常数优化部分:
其中有一段代码是这样的:

for(int i=1;i<=n;i++)
{
   lg[i]=lg[i-1]+((1<<lg[i-1])==i);
   }

其中有 l g [ i ] = m i n { x ∣ i < 2 x } lg[i]=min\left \{ x|i<2^{x} \right \} lg[i]=min{xi<2x}
(giao我终于会在CSDN里面放latex公式了)
这一步动态规划来的很神妙,其实可以理解成位运算
并且应该注意到 l g [ i ] = 0 lg[i]=0 lg[i]=0,所以这里面算到lg[1]的时候就把1左移0位然后看和1相不相等,并且注意到在 i = = 2 l g [ i − 1 ] i==2^{lg[i-1]} i==2lg[i1]时, l g [ i ] + + lg[i]++ lg[i]++所以有最初的推导关系。这一步是常数优化之后会用到。
接下来是 O ( n l g n ) O(nlgn) O(nlgn)的预处理,这一步也很有意思
主要用到的代码是:

fa[cur][i]=fa[fa[cur][i-1]][i-1];

其中 f a [ c u r ] [ i ] 为 c u r 节点的 2 i 级的祖先 fa[cur][i]为cur节点的2^{i}级的祖先 fa[cur][i]cur节点的2i级的祖先
使用 f a [ c u r ] [ i − 1 ] fa[cur][i-1] fa[cur][i1]主要是跳到本节点的 p o w ( 2 , i − 1 ) pow(2,i-1) pow(2,i1)的祖先,然后在 i − 1 i-1 i1次幂求 i i i次幂的祖先,上图:
在这里插入图片描述
然后是最后一部分,俺觉得比较有意思的:
其用到的代码块是:

for(int k=lg[dep[cur]]-1;k>=0;k--)
{
	if(fa[x][k]!=fa[y][k])
	{
	x=fa[x][k],y=fa[y][k];
	}}
return fa[x][0];//LCA

接前文提到的 l g [ i ] lg[i] lg[i]代表的是位运算的事情
这里面可以看作是检查每一位找到那个父节点的位置
举个例子来看,比如说一个 c u r cur cur节点深度的二进制位为

1010100011

(一共有 10 10 10
那个这时这个 k k k就是为 10 10 10
而比较来看 2 10 2^{10} 210应该有 11 11 11

 1010100011
10000000000

而如果 k k k 1 1 1的话就变成了:

1010100011
1000000000

这个时候相当于是在检查最高位,他是咋检查的捏?我们看,从第 0 0 0级到 c u r cur cur所在深度中一定有一个位置是这个公共祖先的位置,所以它相当于是在排除位,先看最高位,他的这个祖先一定是在
[ 0 , 0010100011 ] o r [ 0010100011 , 1010100011 ] [0,0010100011]or[0010100011,1010100011] [0,0010100011]or[0010100011,1010100011]
如果他两个祖先不相等,那就说明应该是在第一个区间里面,如果祖先相等就是在第二个区间里,这很好理解,画个图来看一下
在这里插入图片描述
所以每一位每一位检查,检查到最后 f a [ x ] [ 0 ] fa[x][0] fa[x][0]就是答案
所以最后 r e t u r n   f a [ x ] [ 0 ] ; return \space fa[x][0]; return fa[x][0];
后记:蒟蒻写的不好请各位大佬不吝赐教
附代码:
(神犇的)(我写的还没 A C AC AC就不放了)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct zzz {
    int t, nex;
}e[500010 << 1]; int head[500010], tot;
void add(int x, int y) {
    e[++tot].t = y;
    e[tot].nex = head[x];
    head[x] = tot;
}
int depth[500001], fa[500001][22], lg[500001];
void dfs(int now, int fath) {
    fa[now][0] = fath; depth[now] = depth[fath] + 1;
    for(int i = 1; i <= lg[depth[now]]; ++i)
        fa[now][i] = fa[fa[now][i-1]][i-1];
    for(int i = head[now]; i; i = e[i].nex)
        if(e[i].t != fath) dfs(e[i].t, now);
}
int LCA(int x, int y) {
    if(depth[x] < depth[y]) swap(x, y);
    while(depth[x] > depth[y])
        x = fa[x][lg[depth[x]-depth[y]] - 1];
    if(x == y) return x;
    for(int k = lg[depth[x]] - 1; k >= 0; --k)
        if(fa[x][k] != fa[y][k])
            x = fa[x][k], y = fa[y][k];
    return fa[x][0];
}
int main() {
    int n, m, s; scanf("%d%d%d", &n, &m, &s);
    for(int i = 1; i <= n-1; ++i) {
        int x, y; scanf("%d%d", &x, &y);
        add(x, y); add(y, x);
    }
    for(int i = 1; i <= n; ++i)
        lg[i] = lg[i-1] + (1 << lg[i-1] == i);
    dfs(s, 0);
    for(int i = 1; i <= m; ++i) {
        int x, y; scanf("%d%d",&x, &y);
        printf("%d\n", LCA(x, y));
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值