[牛客13611] 树

树

题目链接：牛客13611 树

Description

给定一颗有 (n) 个节点的树，有 (k) 种颜料给树染色。
一个染色方案是合法的，当且仅当对于所有相同颜色的点对 ((x,y)) ，满足(x)到(y)的路径上所有点的颜色相同（包括(x)和(y)）。
统计方案数，答案对(1e9+7)取模。
数据范围 (1le n,kle 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
", ans); 
  return 0;
}