L C A

知乎算法笔记
学习博客
先初步了解一下

知乎上的板子
倍增LCA

我们用一个数组 f a [ i ] [ k ] fa[i][k] fa[i][k] 存储 i i i 号点的 2 k 2^k 2k 级祖先。(父节点为 1 1 1 级祖先,祖父结点为 2 2 2 级祖先……以此类推)
f a [ c u r ] [ i ] = f a [ f a [ c u r ] [ i − 1 ] ] [ i − 1 ] fa[cur][i] = fa[fa[cur][i - 1]][i - 1] fa[cur][i]=fa[fa[cur][i1]][i1]
这个意思应该是 c u r cur cur 节点的 2 i − 1 2^{i-1} 2i1 级祖先的 2 i − 1 2^{i-1} 2i1 级祖先,也就是 c u r cur cur 2 i 2^i 2i 级祖先

树上两点 u , v u,v u,v 的距离,有公式 d i s u , v = d e p u + d e p v − 2 d e p l c a ( u , v ) dis_{u,v}=dep_{u}+dep_{v}-2dep_{lca(u,v)} disu,v=depu+depv2deplca(u,v)
O ( n l o g n ) 预 处 理 , O ( l o g n ) 查 询 , 空 间 复 杂 度 O ( n l o g n ) O(nlogn)预处理,O(logn)查询,空间复杂度O(nlogn) O(nlogn)O(logn)O(nlogn)
当然,以上都是针对无权树的,如果有权值,可以额外记录一下每个点到根的距离,然后用几乎相同的公式求出。
code:

int Log2[MAXN], fa[MAXN][20], dep[MAXN]; // fa的第二维大小不应小于log2(MAXN)
bool vis[MAXN];
void dfs(int cur, int fath = 0)
{
    if (vis[cur])
        return;
    vis[cur] = true;
    dep[cur] = dep[fath] + 1;
    fa[cur][0] = fath;
    for (int i = 1; i <= Log2[dep[cur]]; ++i)
        fa[cur][i] = fa[fa[cur][i - 1]][i - 1];
    for (int eg = head[cur]; eg; eg = edges[eg].next)
        dfs(edges[eg].to, cur);
}
int lca(int a, int b)
{
    if (dep[a] > dep[b])// 不妨设a的深度小于等于b
        swap(a, b);
    while (dep[a] != dep[b])// 跳到深度相等为止
        b = fa[b][Log2[dep[b] - dep[a]]];// b不断往上跳
    if (a == b)
        return a;
    for (int k = Log2[dep[a]]; k >= 0; k--)
        if (fa[a][k] != fa[b][k])
            a = fa[a][k], b = fa[b][k];
    return fa[a][0];
}
int main()
{
    // ...
    for (int i = 2; i <= n; ++i)// 预处理 log2(i)
        Log2[i] = Log2[i >> 1] + 1;
    // ...
    dfs(s); // 无根树可以随意选一点为根
    // ...
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值