【SCOI2008】着色方案

【SCOI2008】着色方案

Description

有n个木块排成一行，从左到右依次编号为1~n。你有k种颜色的油漆，其中第i 种颜色的油漆足够涂ci 个木块。所有油漆刚好足够涂满所有木块，即 c1+c2+...+ck=n。相邻两个木块涂相同色显得很难看，所以你希望统计任意两个相邻木块颜色不同的着色方案。

Input

第一行为一个正整数k，第二行包含k个整数c1, c2, ... , ck。

Output

输出一个整数，即方案总数模1,000,000,007的结果。

Sample Input

样例1：
3
1 2 3
样例2：
5
2 2 2 2 2
样例3：
10
1 1 2 2 3 3 4 4 5 5

Sample Output

样例1：
10
样例2：
39480
样例3：
85937576

Hint

数据规模：
50%的数据满足：1 <= k <= 5, 1 <= ci <= 3
100%的数据满足：1 <= k <= 15, 1 <= ci <= 5

Source

SCOI，递推 ，动态规划 ，散列

Solution

记忆化搜索，对于每一中涂色的可能性分情况搜索即可

Code

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <map>
#include <vector>
#include <queue>
#define LL long long
#define mod 1000000007
using namespace std;

int k, c[20], x;
LL f[16][16][16][16][16][6], t;
bool vis[16][16][16][16][16][6];

inline LL dp(int a, int b, int c, int d, int e, int k) {
  LL t = 0;
  if (vis[a][b][c][d][e][k]) return f[a][b][c][d][e][k];
  if (a + b + c + d + e == 0) return 1;
  if (a) t += (a - (k == 2)) * dp(a - 1, b, c, d, e, 1);
  if (b) t += (b - (k == 3)) * dp(a + 1, b - 1, c, d, e, 2);
  if (c) t += (c - (k == 4)) * dp(a, b + 1, c - 1, d, e, 3);
  if (d) t += (d - (k == 5)) * dp(a, b, c + 1, d - 1, e, 4);
  if (e) t += e * dp(a, b, c, d + 1, e - 1, 5);
  vis[a][b][c][d][e][k] = 1;
  return f[a][b][c][d][e][k] = (t % mod);
}

int main() {
  freopen("2027.in", "r", stdin);
  freopen("2027.out", "w", stdout);
  scanf("%d", &k);
  for (int i = 1; i <= k; ++i) scanf("%d", &x), c[x]++;
  printf("%lld\n", dp(c[1], c[2], c[3], c[4], c[5], 0));
  return 0;
}