组昂树

组昂树用于统计每个节点的子树的信息。复杂度为\(O(n\log n)\)。

先将原树轻重路径剖分，之后遍历树。solve(u)表示统计u结点子树的答案，考虑以下过程：先调用solve(轻儿子)，且不保留计算结果。然后调用solve(重儿子)，保留计算结果。最后再计算所有轻儿子的子树的答案和自己的答案，更新答案。最后考虑是否保留计算结果。

calc(u, type) {
    contribute f[u] <- type;
    for v = each son of u
        if mark[v] = False
            calc(v, type);
}

solve(u, keep) {
    hson = find the heaviest son of u;
    for v = each son of u
        if v != hson
            solve(v, False);
    if hson exists {
        solve(hson, True);
        mark[hson] = True;
    }
    calc(u, 1);
    if hson exists 
        mark[hson] = False;
    if keep = False
        calc(u, -1);
}