Description
Input
Output
输出仅一行,包含一个正整数,表示合法质量序列的数目对 10^9+7取模的结果。
Sample Input
1 < 2
1 < 3
2 < 4
1 = 5
Sample Output
HINT
不同的合法序列共5个,如下所示:
题解
做之前偷瞄了一眼标签,想的时候就直接往上靠了,并不是很难想。
首先一个值得肯定的条件就是,如果出现 $u$ '=' $v$ ,显然 $u$ 和 $v$ 可以缩成一个点。用并查集辅助。
若我们将 $u$ '<' $v$ 看成是一条 $u->v$ 的有向边,注意到题目说的对于每个点,只有一个“前驱”(缩点后也只有一个前驱),所以说如何合法的话这些节点和边构成了一些树形图的森林。
我们建立 $0$ 节点,来使森林联通,遍历整棵树,除去不合法的情况。
现在问题就变成了给你一棵树,让你生成一个包含所有数且两个数之间用 '=' 或 '<' 号连接的序列。保证父亲总比儿子先在序列中出现,同一棵子树的内部用 '<' 连接;不同儿子的子树间用 '=' 或 '<' 连接。问方案数。
考虑这样一个问题:若只能用 '<' 连接。这个问题将极度简化,我们记 $f_u$ 表示在以 $u$ 节点为根的子树中的方案数。$f_u$ 可起到临时存中间变量的作用。
枚举儿子节点 $v$ 的时候,我们用 $tol$ 表示已处理过的子树的总大小: $$f_u = f_u*f_v*C_{tol+f_v}^{f_v}$$
现在考虑原问题。由于子树间的等号关系不好处理,我们可以将其放到状态中,我们记 $f_{u, k}$ 为在以 $u$ 为根的子树中生成的序列含有 $k$ 个 '<' 的方案数。
现在我们还是考虑 $dp$ 。
假设已处理过的子树总 $size$ 为 $a$ ,记录下来的 $dp$ 值存在 $p_i$ 中,正在处理的子树的 $size$ 为 $b$ ,记录下来的 $dp$ 值存在 $q_i$ 中。
由题,合并后的 '<' 的个数介于 $max(a, b)$ 和 $a+b$ 之间。
我们枚举 $i <= a, j <= b$,同理合并后的 '<' 的个数介于 $max(i, j)$ 和 $i+j$ 之间。现在相当于将 $i$ 个白球, $j$ 个黑球放入 $k$ 个盒子中,且同个盒子不能有相同颜色的球,盒子不能空。
$$f_{u, k} += sum_{i = 1}^a sum_{j = 1}^b p_i*q_j*C_k^i*C_i^{j-(k-i)}$$
其中 $C_k^i$ 表示在 $k$ 个盒子中选出 $i$ 个放白球,因为所有盒子都要放球,所以剩下的 $k-i$ 个盒子必定放黑球,剩下 $j-(k-i)$ 个黑球要放在 $i$ 个放白球的盒子中。
1 //It is made by Awson on 2018.1.2 2 #include <set> 3 #include <map> 4 #include <cmath> 5 #include <ctime> 6 #include <queue> 7 #include <stack> 8 #include <cstdio> 9 #include <string> 10 #include <vector> 11 #include <cstdlib> 12 #include <cstring> 13 #include <iostream> 14 #include <algorithm> 15 #define LL long long 16 #define LD long double 17 #define Max(a, b) ((a) > (b) ? (a) : (b)) 18 #define Min(a, b) ((a) < (b) ? (a) : (b)) 19 using namespace std; 20 const int N = 100; 21 const int MOD = 1e9+7; 22 23 int fa[N+5], n, m, pre[N+5], u, v, vis[N+5], rt[N+5], C[N+5][N+5], f[N+5][N+5], size[N+5]; 24 char ch[5]; 25 struct tt { 26 int to, next; 27 }edge[N+5]; 28 int path[N+5], top; 29 30 int find(int r) { 31 return fa[r] ? fa[r] = find(fa[r]) : r; 32 } 33 void add(int u, int v) { 34 edge[++top].to = v; 35 edge[top].next = path[u]; 36 path[u] = top; 37 } 38 void dfs(int u) { 39 int g[N+5], flag = 0; 40 for (int I = path[u]; I; I = edge[I].next) { 41 memset(g, 0, sizeof(g)); dfs(edge[I].to); 42 if (flag) { 43 for (int i = 1; i <= size[u]; i++) 44 if (f[u][i]) 45 for (int j = 1; j <= size[edge[I].to]; j++) 46 if (f[edge[I].to][j]) 47 for (int k = Max(i, j); k <= i+j; k++) 48 g[k] = (g[k]+(LL)f[u][i]*f[edge[I].to][j]%MOD*C[k][i]%MOD*C[i][j-(k-i)]%MOD)%MOD; 49 size[u] += size[edge[I].to]; 50 for (int i = 1; i <= size[u]; i++) f[u][i] = g[i]; 51 }else { 52 flag = 1, size[u] += size[edge[I].to]; 53 for (int i = 1; i <= size[u]; i++) f[u][i] = f[edge[I].to][i]; 54 } 55 } 56 if (!flag) f[u][0] = 1; 57 ++size[u]; 58 for (int i = size[u]; i; i--) f[u][i] = f[u][i-1]; 59 } 60 bool judge(int u) { 61 if (vis[u]) return false; vis[u] = 1; 62 for (int i = path[u]; i; i = edge[i].next) if (!judge(edge[i].to)) return false; 63 return true; 64 } 65 void work() { 66 scanf("%d%d", &n, &m); 67 for (int i = 0; i <= n; i++) { 68 C[i][0] = 1; 69 for (int j = 1; j <= i; j++) C[i][j] = (C[i-1][j-1]+C[i-1][j])%MOD; 70 } 71 for (int i = 1; i <= m; i++) { 72 scanf("%d%s%d", &u, ch, &v); 73 if (ch[0] == '=') { 74 int p = find(u), q = find(v); 75 if (p != q) fa[p] = q, rt[p] = 1, pre[q] = Max(pre[q], pre[p]); 76 }else pre[v] = u; 77 } 78 for (int i = 1; i <= n; i++) if (!rt[i]) add(find(pre[i]), i); 79 for (int i = 0; i <= n; i++) if (!vis[i]) if (!judge(i)) {printf("0 "); return; } 80 dfs(0); 81 int ans = 0; 82 for (int i = 1; i <= size[0]; i++) ans = (ans+f[0][i])%MOD; 83 printf("%d ", ans); 84 } 85 int main() { 86 work(); 87 return 0; 88 }