树链剖分算法整理

树链剖分

树链剖分可以把树分成若干条链，从而维护树上的路径信息。本质思想是把树剖成可以用线性结构存储的结构，然后可以数据结构维护。

分为三种：重链剖分、长链剖分、实链剖分。以下以重链剖分为主。

重链剖分可以将树上的任意一条路径划分成不超过$O(logn)$条连续的链，每条链上的点深度互不相同（即是自底向上的一条链，链上所有点的 LCA 为链的一个端点）。

重链剖分还能保证划分出的每条链上的节点 DFS 序连续，因此可以方便地用一些维护序列的数据结构（如线段树）来维护树上路径的信息。

相比之下，树链剖分能解决的很多问题用树上差分也能解决，两者的关系近似于于线段树和差分的关系。

定义表：

重子节点：其子节点中子树最大的子结点。如果有多个子树最大的子结点，取其一。如果没有子节点，就无重子节点。

轻子节点：除了重子节点的所有子节点。

重边：从任意节点到重子节点的边。

轻边：从任意节点到轻子节点的边。

重链：若干条首尾衔接的重边，也包括了落单的节点。

如下图 给出了具体的解释（图源OI-WiKi）

为了剖出上述重链，我们需要维护如下信息：

• $fa[x]$：节点x在树上的父亲。
• $dep[x]$：节点x在树上的深度。
• $siz[x]$：节点x的子树节点个数。
• $son[x]$：节点x的重儿子下标。
• $top[x]$：节点x所在的重链的顶部（深度最小）节点。
• $dfn[x]$：节点x的DFS序，即相当于用数据结构维护时的原始下标。
• $rnk[x]$：$dfn$反向映射，$rnk[dfn[x]]=x$。

由两次DFS即可处理出这些信息，第一次处理前四项，后一次处理后四项。

处理代码：

void dfs1(int u, int f) {
   
    fa[u] = f;
    dep[u] = dep[f] + 1;
    siz[u] = 1;
    int maxsz = -1;
    for (auto v : G[u]) {
   
        if (v == f) continue;
        dfs1(v, u);
        siz[u] += siz[v];
        if (siz[v] > maxsz) {
   
            maxsz = siz[v];
            son[u] = v;
        }
    }