树
题目链接:牛客13611 树
Description
给定一颗有 \(n\) 个节点的树,有 \(k\) 种颜料给树染色。
一个染色方案是合法的,当且仅当对于所有相同颜色的点对 \((x,y)\) ,满足\(x\)到\(y\)的路径上所有点的颜色相同(包括\(x\)和\(y\))。
统计方案数,答案对\(1e9+7\)取模。
数据范围 \(1\le n,k\le 300\)
Solution
这是一道结论题。
如果直接做,显然不好做,我们考虑切换视角。
在树上求点对路径,其实可以通过\(dfs\)序转换到链上。
我们先跑一次\(dfs\),记下每个点的\(dfs序\) \(dfn[x]\) 。
我们会发现,如果当前遍历到了点 \(x\) ,那么 \(x\) 的祖先们肯定都已经遍历过了,它的兄弟们有部分被遍历过。
我们规定\(x\)和\(y\)的排列顺序是按照\(dfn[y] < dfn[x]\)排列的,那么我们枚举\(x\)的时候,\(y\)已经上色了。
如果想要遍历到\(y\),那么\(x\)无论如何都要经过它的父亲,所以\(x\)一定与\(father[x]\)同色。
因此,我们发现,这个树的形态我们并不关心,真正重要的只是\(n\)和\(k\)。
好,接下来,我们定义\(f[i][j]\)表示按照\(dfs序\)排列,前\(i\)个节点染了\(j\)种颜色的方案数。
考虑如何转移:
- 第\(i\)个节点染这\(j\)种颜色中的一种,那么方案数为 \(f[i-1][j]\),因为它的颜色一定与它父亲颜色相同;
- 第\(i\)个节点染新的一种颜色,那么方案数为 \(f[i-1][j-1]*(k-(j-1))\),因为还剩余 \(k-j+1\) 种颜色供 \(i\) 选择。
显然边界条件是 \(f[0][0]=1\)。
答案即为 \(\sum_{i=1}^{k} f[n][i]\)。
时间复杂度 \(O(nk)\) ,可以通过本题。
PS:为啥这题的数据范围这么小啊……
Code
// Author: wlzhouzhuan
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define rint register int
#define rep(i, l, r) for (rint i = l; i <= r; i++)
#define per(i, l, r) for (rint i = l; i >= r; i--)
#define mset(s, _) memset(s, _, sizeof(s))
#define pb push_back
#define pii pair <int, int>
#define mp(a, b) make_pair(a, b)
#define Each(i) for (rint i = head[u]; i; i = edge[i].nxt)
inline int read() {
int x = 0, neg = 1; char op = getchar();
while (!isdigit(op)) { if (op == '-') neg = -1; op = getchar(); }
while (isdigit(op)) { x = 10 * x + op - '0'; op = getchar(); }
return neg * x;
}
inline void print(int x) {
if (x < 0) { putchar('-'); x = -x; }
if (x >= 10) print(x / 10);
putchar(x % 10 + '0');
}
const int mod = 1e9 + 7;
const int N = 301;
int f[N][N];
int n, k;
int main() {
n = read(), k = read();
f[0][0] = 1;
for (rint i = 1; i <= n; i++) {
for (rint j = 1; j <= k; j++) {
f[i][j] = (f[i - 1][j] + 1ll * f[i - 1][j - 1] * (k - j + 1) % mod) % mod;
}
}
int ans = 0;
for (rint i = 1; i <= k; i++) {
ans = (ans + f[n][i]) % mod;
}
printf("%d\n", ans);
return 0;
}