长链剖分介绍

什么是长链剖分?

长链剖分是树链剖分的一种,与重链剖分不同,它选择子树深度最大的儿子作为长儿子‌,这样做的目的是优化处理与树深度相关的问题,比如查询k级祖先等。

长链剖分的主要性质

1、每个节点属于且仅属于一条长链;2、所有长链的节点和为N;3、k级祖先所在的长链长度必须大于等于k‌。

长链剖分的实现

定义数组来存储深度、长儿子、链顶等信息;第一次DFS(dfs1)负责计算深度、链长和长儿子;第二次DFS(dfs2)确定链顶。例如,使用dep数组记录深度,hson数组记录长儿子,top数组记录链顶。

C++长链剖分模板程序

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+5;

vector<int> G[N];    // 邻接表存树
int dep[N], len[N];   // dep[u]:u的深度,len[u]:u所在长链长度
int fa[N], hson[N];   // fa[u]:父节点,hson[u]:长儿子
int top[N];           // top[u]:u所在长链的链顶

// 第一次DFS:计算深度、长儿子、链长
void dfs1(int u, int father) {
    dep[u] = dep[father] + 1;
    fa[u] = father;
    hson[u] = 0;
    len[u] = 1;
    
    for(int v : G[u]) {
        if(v == father) continue;
        dfs1(v, u);
        // 更新长儿子(选择深度最大的子树)
        if(len[v] + 1 > len[u]) {
            len[u] = len[v] + 1;
            hson[u] = v;
        }
    }
}

// 第二次DFS:确定链顶
void dfs2(int u, int tp) {
    top[u] = tp;
    if(hson[u]) {
        dfs2(hson[u], tp);  // 长儿子继承链顶
        for(int v : G[u]) {
            if(v == fa[u] || v == hson[u]) continue;
            dfs2(v, v);     // 其他儿子新建链
        }
    }
}

int main() {
    int root = 1;           // 根据题目确定根节点
    dfs1(root, 0);
    dfs2(root, root);
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值