BZOJ 1079 [记忆化搜索]

1079: [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
3
1 2 3
Sample Output
10
100%的数据满足：1 <= k <= 15, 1 <= ci <= 5

很显然的DP $5^{15}$ 太大
$15^5$就可以
用res[a][b][c][d][e][last] 表示 还可以填1个木块的油漆有a个，还可以填2个木块的油漆有b个…..上次用的是还可以涂last个的木块。
得到了转移方程

inline ll dp(int a, int b, int c, int d, int e,int last) {
  if (res[a][b][c][d][e][last] != -1)
    return res[a][b][c][d][e][last];
  if ((a | b | c | d | e) == 0)
    return res[a][b][c][d][e][last] = 1;
  ll ans = 0;
  if (a) ans = ans + ll(dp(a - 1, b, c, d, e, 1) * (a - (last == 2))), ans = ans % MOD;
  if (b) ans = ans + ll(dp(a + 1, b - 1, c, d, e, 2) * (b - (last == 3))), ans = ans % MOD;
  if (c) ans = ans + ll(dp(a, b + 1, c - 1, d, e, 3) * (c - (last == 4))), ans = ans % MOD;
  if (d) ans = ans + ll(dp(a, b, c + 1, d - 1, e, 4) * (d - (last == 5))), ans = ans % MOD;
  if (e) ans = ans + ll(dp(a, b, c, d + 1, e - 1, 5) * e), ans = ans % MOD;
  return res[a][b][c][d][e][last] = ans;
}

代码：

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <cstring>
#define MOD 1000000007
using namespace std;

typedef long long ll;

int n, k, ans;
int c[15];
ll res[17][17][17][17][17][6];

inline ll dp(int a, int b, int c, int d, int e,int last) {
  if (res[a][b][c][d][e][last] != -1)
    return res[a][b][c][d][e][last];
  if ((a | b | c | d | e) == 0)
    return res[a][b][c][d][e][last] = 1;
  ll ans = 0;
  if (a) ans = ans + ll(dp(a - 1, b, c, d, e, 1) * (a - (last == 2))),
    ans = ans % MOD;
  if (b) ans = ans + ll(dp(a + 1, b - 1, c, d, e, 2) * (b - (last == 3))),
    ans = ans % MOD;
  if (c) ans = ans + ll(dp(a, b + 1, c - 1, d, e, 3) * (c - (last == 4))),
    ans = ans % MOD;
  if (d) ans = ans + ll(dp(a, b, c + 1, d - 1, e, 4) * (d - (last == 5))),
    ans = ans % MOD;
  if (e) ans = ans + ll(dp(a, b, c, d + 1, e - 1, 5) * e),
    ans = ans % MOD;
  return res[a][b][c][d][e][last] = ans;
}

int main(void) {
  freopen("1.in", "r", stdin);
  scanf("%d", &n);
  memset(res, -1, sizeof(res));
  memset(c, 0, sizeof(c));
  for(int i = 0; i < n; i++) {
    scanf("%d", &k);
    c[k]++;
  }
  cout << dp(c[1], c[2], c[3], c[4], c[5], 0) << endl;
  return 0;
}