P8867 题解

Chy12321

2022-12-06 19:18:16

Solution

只有 B 国炸毁了图的割边,才会使得图不连通,进而可能会导致军营不连通。也就是说,A 国可以随意地看守或不看守不是割边的边。因此想到 [边双缩点](https://www.luogu.com.cn/problem/P8436) 后树形 DP。 为什么边双缩点后会形成一棵树呢? 题目保证了给定的图连通,那么缩点后的图也必然连通,而如果有多个“双连通分量”构成了环,就不符合双连通分量的定义了,即这些首尾相连构成环的“双连通分量”应该被划在同一个双连通分量中。 因此,缩点后形成的图连通且无环,也就形成了一棵树。 已经缩了点,再思考:究竟要在缩点后形成的树上求什么? 令 $V_u$ 表示双连通分量 $u$ 中的点数,$E_u$ 表示双连通分量 $u$ 中的边数,若有 $n$ 个双连通分量,则问题转化为: 给定一棵无根树,每个结点有 $2^{E_u}$ 种不建造军营的方案和 $(2^{V_u + E_u} - 2^{E_u})$ 种建造军营的方案。求共有多少种建造军营的方案(不能不建)。 这里假定 $1$ 号结点为树根。 令 $f(u, 0/1)$ 表示以 $u$ 为根的子树中没有/有军营的方案数。 发现每种状态所涵盖的情况过多,根本不好转移。 这时,有两种思路: - 增加状态数量。 - 对状态增添限制。 我选择的是后者。 令 $f(u, 0/1)$ 表示以 $u$ 为根的子树中没有/有军营的方案数,若有军营,则所有的军营必须通过已经派兵看守的边与 $u$ 连通。 在想转移之前,为了防止做无用功,最好先想想该如何统计答案。 对于每个结点 $u$,我们强制 $u$ 子树外的所有点都不建军营,同时强制不选 $fa_u \to u$ 的边,再累加方案数,即可保证不重不漏(具体可看下图)。 ![](https://cdn.luogu.com.cn/upload/image_hosting/ufhysx5o.png) 令 $s(u)$ 表示以 $u$ 为根节点的子树内边数,即 $s(u) = E_u + \sum\limits_{v \in son(u)} [s(v) + 1]$,则有 $ans \leftarrow f(u, 1) \times 2^{s(1) - s(u) - 1}$。 特殊地,对于 $1$ 号结点,不存在 $fa_1 \to 1$ 的边,此时 $ans \leftarrow f(1, 1)$。 明确了答案如何统计,接下来考虑转移: 显然地,$f(u, 0) = 2^{E_u} \times \prod\limits_{v \in son(u)} 2f(v, 0)$,难点在 $f(u, 1)$ 的转移上。 考虑每新增一个子节点 $v$ 对 $f(u, 1)$ 产生的贡献。 若到新增前都还未建造一个军营,则以 $v$ 为根的子树中必须有军营,即 $f(u, 1) \leftarrow f(u, 0) \times f(v, 1)$。 若到新增前已经建造过军营,则以 $v$ 为根的子树中有没有军营皆可,且当以 $v$ 为根的子树中没有军营时,$v$ 点是否与 $u$ 点连通皆可,即 $f(u, 1) \leftarrow f(u, 1) \times [2f(v, 0) + f(v, 1)]$。 综上,$f(u, 1) \leftarrow f(u, 0) \times f(v, 1) + f(u, 1) \times [2f(v, 0) + f(v, 1)]$。 初始时,$f(u, 0) = 2^{E_u}$,$f(u, 1) = 2^{V_u + E_u} - 2^{E_u}$。 代码: ```cpp #include <bits/stdc++.h> using namespace std; typedef long long ll; constexpr int N = 500100, M = 1000100, MOD = 1e9 + 7; int n, m, p; int tot, tot2, head[N], head2[N]; int cnt, top, stk[N], dfn[N], low[N], bel[N]; int deg[N], V[N], E[N], s[N]; bool ins[N]; ll ans, f[N][2]; struct Edge { int to, nxt; } e[M << 1], e2[M << 1]; void add(int u, int v) { e[++tot] = Edge{v, head[u]}; head[u] = tot; } void add2(int u, int v) { e2[++tot2] = Edge{v, head2[u]}; head2[u] = tot2; } void tarjan(int u, int fa) { dfn[u] = low[u] = ++cnt, ins[stk[++top] = u] = 1; for (int i = head[u], v; i; i = e[i].nxt) { v = e[i].to; if (v == fa) continue; if (!dfn[v]) tarjan(v, u), low[u] = min(low[u], low[v]); else if (ins[v]) low[u] = min(low[u], dfn[v]); } if (dfn[u] == low[u]) { p++; int x; do { // 因为不需要知道每个边双连通分量里都有哪些点,只记录每个点属于哪个边双连通分量即可。 ins[x = stk[top--]] = 0, bel[x] = p; V[p]++; // 累加该边双连通分量内点数 } while (x != u); } } ll qp(ll base, int e) { // 快速幂 ll res = 1; while (e) { if (e & 1) res = res * base % MOD; base = base * base % MOD; e >>= 1; } return res; } void dfs(int u, int fa) { // dfs 计算 s[] s[u] = E[u]; for (int i = head2[u], v; i; i = e2[i].nxt) { v = e2[i].to; if (v == fa) continue; dfs(v, u); s[u] += s[v] + 1; } } void dp(int u, int fa) { // 树形 DP for (int i = head2[u], v; i; i = e2[i].nxt) { v = e2[i].to; if (v == fa) continue; dp(v, u); // 状态转移 f[u][1] = (f[u][1] * (((f[v][0] << 1) + f[v][1]) % MOD) % MOD + f[u][0] * f[v][1] % MOD) % MOD; f[u][0] = f[u][0] * ((f[v][0] << 1) % MOD) % MOD; } // 统计答案 if (u == 1) ans = (ans + f[u][1]) % MOD; // 特判 1 号结点的特殊情况 else ans = (ans + f[u][1] * qp(2, s[1] - s[u] - 1)) % MOD; } int main() { ios::sync_with_stdio(0); cin.tie(nullptr), cout.tie(nullptr); cin >> n >> m; while (m--) { int u, v; cin >> u >> v; add(u, v), add(v, u); } tarjan(1, 0); // 边双缩点 for (int u = 1; u <= n; u++) { for (int i = head[u], v; i; i = e[i].nxt) { v = e[i].to; if (bel[u] != bel[v]) add2(bel[u], bel[v]); // 如果属于两个不同的边双连通分量,则将这两个边双连通分量连边 else E[bel[u]]++; // 否则该双连通分量内边数 + 1 } } for (int i = 1; i <= p; i++) { E[i] >>= 1; // 因为是无向边,每一条边会累加 2 次,故 E[i] 需要除以 2 // 赋初值 f[i][0] = qp(2, E[i]); f[i][1] = qp(2, V[i] + E[i]) - f[i][0]; } dfs(1, 0); dp(1, 0); cout << ans; return 0; } ``` --- upd on 2022.12.18 感谢 @一只绝帆 指出了一处笔误。 upd on 2023.5.15 针对 @__vector__ 指出的错误修改了部分内容。