Problem - K Russian Dolls on the Christmas Tree - Codeforces （树上启发式合并）

Problem - K Russian Dolls on the Christmas Tree - Codeforces

题意

有一棵 $n$ 个节点的树，节点编号为 $1-n$，$1$ 号节点是根。问以每个节点为根的子树中，有多少个连续区间。

解法

子树？离线？立即推：树上启发式合并！

• 统计子树每个编号的点出现的次数，同时维护答案；
• 如果编号为 $u$ 的点从无到有，就看编号为 $u-1$ 和 $u+1$ 的点的出现情况，如果都出现了，就说明两个区间合并为一个，区间数减一，如果都没有出现，则新增一个区间。
• 如果编号为 $u$ 的点从有到无，就看编号为 $u-1$ 和 $u+1$ 的点的出现情况，如果都出现了，就说明一个区间分裂成两个，区间数加一，如果都没有出现，则区间数减一。

解法

#pragma region
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <vector>
using namespace std;
typedef long long ll;
#define tr t[root]
#define lson t[root << 1]
#define rson t[root << 1 | 1]
#define rep(i, a, n) for (int i = a; i <= n; ++i)
#define per(i, a, n) for (int i = n; i >= a; --i)
#pragma endregion
const int maxn = 2e5 + 5;
int n;
vector<int> g[maxn];
int son[maxn], sz[maxn];
int res[maxn], flag;
int cnt[maxn], ans;
void init() {
    rep(i, 1, n) g[i].clear(), son[i] = 0;
}
void dfs1(int u, int f) {
    sz[u] = 1;
    for (auto v : g[u]) {
        if (v == f) continue;
        dfs1(v, u);
        sz[u] += sz[v];
        if (sz[v] > sz[son[u]]) son[u] = v;
    }
}
void count(int u, int f, int val) {
    cnt[u] += val;
    if (cnt[u - 1] && cnt[u + 1]) ans -= val;
    if (!cnt[u - 1] && !cnt[u + 1]) ans += val;
    for (auto v : g[u]) {
        if (v == f || v == flag) continue;
        count(v, u, val);
    }
}
void dfs(int u, int f, bool keep) {
    for (auto v : g[u]) {
        if (v == f || v == son[u]) continue;
        dfs(v, u, 0);
    }
    if (son[u]) {
        dfs(son[u], u, 1);
        flag = son[u];
    }
    count(u, f, 1);
    res[u] = ans;
    flag = 0;
    if (!keep) count(u, f, -1);
}
int main() {
    int T;
    scanf("%d", &T);
    rep(cas, 1, T) {
        scanf("%d", &n);
        init();
        rep(i, 1, n - 1) {
            int u, v;
            scanf("%d%d", &u, &v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        dfs1(1, 0);
        dfs(1, 0, 0);
        printf("Case #%d:", cas);
        rep(i, 1, n) printf(" %d", res[i]);
        puts("");
    }
}