LCA模板(倍增)

LCA(Lowest Common Ancestor): 树中两个节点深度最大的公共祖先.

思想: 类似ST表,打倍增。预处理出每个节点的深度后,不是一个个跳,而是跳着跳。一次跳2的k次方.
核心: f[i][j]表示i向上跳2^j的父节点。 f[i][j] = f[f[i][j-1]][j-1]
时间复杂度: 预处理O(nlogn),单次查询O(logn)

fa[i][j]:节点i向上跳2^j的父节点
lg[i]:log(i)+1
dep[i]:深度

写法:
1.dfs预处理深度和fa数组,预处理lg数组,lg[x]表示log(x)+1
2.求LCA(x,y)
[1] 不妨假设x的深度 >= y的深度,不是则交换
[2] 让x向上跳,跳到和y同一深度。用到二进制拆位的思想,从最大的能跳的k开始枚举。
[3] x、y一起跳,若fa[x][k] == fa[y][k],是公共祖先,但是不一定是LCA。所以我们为了避免误判,跳到LCA下边一层,避免误判。

例题

模板:

#include<bits/stdc++.h>
using namespace std;
const int N = 5e5+10;
int n,m,k,T;
int e[N<<1],ne[N<<1],h[N<<1],idx = 0;
void add(int a,int b)
{
    e[idx] = b,ne[idx] = h[a],h[a] = idx++;
}
int fa[N][17],depth[N],lg[N];
void init()
{
    for(int i=1;i<=n;++i) lg[i] = lg[i-1] + (1<<lg[i-1]==i);
}
void dfs(int cur,int father)
{
    fa[cur][0] = father; depth[cur] = depth[father]+1;
    for(int i=1;i<=lg[depth[cur]];++i)
    {
        fa[cur][i] = fa[fa[cur][i-1]][i-1];
    }
    for(int i=h[cur];~i;i=ne[i])
    {
        int j = e[i];
        if(j!=father) dfs(j,cur);
    }
}
int LCA(int x,int y)
{
    if(depth[x] < depth[y]) swap(x,y); //不妨假设x深度更大
    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) //跳到LCA下边一层
    {
        if(fa[x][k] != fa[y][k])
        {
            x = fa[x][k],y = fa[y][k];
        }
    }
    return fa[x][0];
}
int root;
int main(void)
{
    memset(h,-1,sizeof(h));
    scanf("%d%d%d",&n,&m,&root);
    init();
    n -- ;
    while(n--)
    {
        int x,y; scanf("%d%d",&x,&y);
        add(x,y),add(y,x);
    }
    dfs(root,0);
    while(m--)
    {
        int x,y; scanf("%d%d",&x,&y);
        int tmp = LCA(x,y);
        printf("%d\n",tmp);
        // if(x == tmp) printf("1\n");
        // else if(y==tmp) printf("2\n");
        // else printf("0\n");
    }
    return 0;
}
  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值