月之暗面
[Link](D-月之暗面_牛客练习赛97 (nowcoder.com))
题意
给一颗 n n n个结点的数染色,普通染色 x x x种,特殊染色 y y y种,普通颜色无限制,两个相邻结点不能染相同的特殊颜色
求染色方案数
思路
先按当前结点普通颜色还是特殊颜色来分类,记 f [ u , 0 ] : 以 u 为 根 且 u 填 普 通 颜 色 的 方 案 数 , f [ u , 1 ] : 以 u 为 根 且 u 填 特 殊 颜 色 的 方 案 数 f[u,0]:以u为根且u填普通颜色的方案数,f[u,1]:以u为根且u填特殊颜色的方案数 f[u,0]:以u为根且u填普通颜色的方案数,f[u,1]:以u为根且u填特殊颜色的方案数,互相独立可以相加,关键在于怎么不从不漏的划分方案。
考虑维护方案对于一个点 u u u与它相邻的非父亲结点 v i v_i vi是相互独立的互不干扰符合乘法原理。
分开来看对于 u u u选 0 0 0子树无限制转移为 f [ u , 0 ] ∗ = f [ v i , 1 ] + f [ v i , 0 ] f[u,0]*=f[v_i,1] +f[v_i,0] f[u,0]∗=f[vi,1]+f[vi,0],对于 u u u选 1 1 1的子树当前根会限制这个子树选 1 1 1的情况, f [ v i ] [ 1 ] f[v_i][1] f[vi][1]中 v i v_i vi选 1 1 1一共 y y y种不同颜色,且无本质差别,由于 u u u的存在 v i v_i vi只能选 y − 1 y-1 y−1种颜色,因此转移为 f [ u , 1 ] ∗ = f [ v i , 0 ] + f [ v i , 1 ] × y − 1 y f[u,1]*=f[v_i,0]+f[v_i,1]\times \frac{y-1}{y} f[u,1]∗=f[vi,0]+f[vi,1]×yy−1
Code
#include <bits/stdc++.h>
#define x first
#define y second
#define debug(x) cout<<#x<<":"<<x<<endl;
using namespace std;
typedef long double ld;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
typedef unsigned long long ULL;
const int N = 1e6 + 10, M = 2 * N, INF = 0x3f3f3f3f, mod = 998244353;
const double eps = 1e-8, pi = acos(-1), inf = 1e20;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
int h[N], e[M], ne[M], w[M], idx;
void add(int a, int b, int v = 0) {
e[idx] = b, w[idx] = v, ne[idx] = h[a], h[a] = idx ++;
}
int n, m, k;
int x, y;
LL f[N][2];
int qmi(int a, int b) {
int res = 1;
while (b) {
if (b & 1) res = (LL) res * a % mod;
a = (LL) a * a % mod;
b >>= 1;
}
return res;
}
void dfs(int u, int fa) {
f[u][0] = x, f[u][1] = y;
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
if (j == fa) continue ;
dfs(j, u);
f[u][0] = f[u][0] * (f[j][0] + f[j][1]) % mod;
f[u][1] = f[u][1] * (f[j][0] + f[j][1] * k % mod) % mod;
}
}
int main() {
ios::sync_with_stdio(false), cin.tie(0);
cin >> n >> x >> y;
memset(h, -1, sizeof h);
k = (LL)(y - 1) * qmi(y, mod - 2) % mod;
for (int i = 1; i < n; i ++) {
int u, v; cin >> u >> v;
add(u, v), add(v, u);
}
dfs(1, -1);
cout << (f[1][0] + f[1][1]) % mod << '\n';
return 0;
}
因为每个点都要选一种颜色,因此 f [ u , 0 ] : 以 u 为 根 且 u 填 了 一 个 普 通 颜 色 的 方 案 数 , f [ u , 1 ] : 以 u 为 根 且 u 了 一 个 填 特 殊 颜 色 的 方 案 数 f[u,0]:以u为根且u填了一个普通颜色的方案数,f[u,1]:以u为根且u了一个填特殊颜色的方案数 f[u,0]:以u为根且u填了一个普通颜色的方案数,f[u,1]:以u为根且u了一个填特殊颜色的方案数,
则转移时候 f [ u , 0 ] ∗ = f [ v i , 0 ] ∗ x + f [ v i , 1 ] ∗ y f[u,0]*=f[v_i,0]*x+f[v_i,1]*y f[u,0]∗=f[vi,0]∗x+f[vi,1]∗y,由于 u u u选了一个因此 v i v_i vi只能选 y − 1 y-1 y−1个特殊颜色,所以 f [ u , 1 ] ∗ = f [ v i , 0 ] ∗ x + f [ v i , 1 ] ∗ ( y − 1 ) f[u,1]*=f[v_i,0]*x+f[v_i,1]*(y-1) f[u,1]∗=f[vi,0]∗x+f[vi,1]∗(y−1)
Code
#include <bits/stdc++.h>
#define x first
#define y second
#define debug(x) cout<<#x<<":"<<x<<endl;
using namespace std;
typedef long double ld;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
typedef unsigned long long ULL;
const int N = 1e6 + 10, M = 2 * N, INF = 0x3f3f3f3f, mod = 998244353;
const double eps = 1e-8, pi = acos(-1), inf = 1e20;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
int h[N], e[M], ne[M], w[M], idx;
void add(int a, int b, int v = 0) {
e[idx] = b, w[idx] = v, ne[idx] = h[a], h[a] = idx ++;
}
int n, m, k;
int x, y;
LL f[N][2];
void dfs(int u, int fa) {
f[u][0] = 1, f[u][1] = 1;
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
if (j == fa) continue ;
dfs(j, u);
f[u][0] = (f[u][0] * (f[j][0] * x % mod + f[j][1] * y % mod) % mod) % mod;
f[u][1] = (f[u][1] * (f[j][0] * x % mod + f[j][1] * (y - 1) % mod) % mod) % mod;
}
}
int main() {
ios::sync_with_stdio(false), cin.tie(0);
cin >> n >> x >> y;
memset(h, -1, sizeof h);
for (int i = 1; i < n; i ++) {
int u, v; cin >> u >> v;
add(u, v), add(v, u);
}
dfs(1, -1);
cout << (f[1][0] * x + f[1][1] * y) % mod << '\n';
return 0;
}