原題地址首先,看這么小的范圍就知道,數學方法肯定搞不了……又想不到其它模型……只能用狀壓硬搞了囧……
問題是,5^15穩T,如果能倒過來,15^5,就不會T了。
可以發現,C值相同的顏色本質上是一樣的……因此,只需要保存目前C值為1、2、3、4、5的顏色各有多少種就行了囧……(當然在過程中還會出現C值為0的,即用完的顏色,不過0、1、2、3、4、5的和是顏色總數,而且從下面可以看出,C值為0的確實“木有用”);
設F[s1][s2][s3][s4][s5][v]為涂完前若干個木塊(這個個數可以通過s1~s5算出,不過我們并不需要它囧……)后,C值為1~5的顏色各有s1~s5種,且這若干個中的
最后一個涂的顏色還剩的C值為v(顯然0<=v<=4)。
邊界:F[S[1]][S[2]][S[3]][S[4]][S[5]][0]=1,其余為0(S[i]為一開始C值為i的顏色種數)。
計算F時,前推后(注意順序,是按照S[5]逆序最先,再依次是S[4]~S[1],都是逆序,v可任意定序),枚舉下一個木塊的顏色是現在還剩多少的,如果它與目前的這個(最后一個)剩的相同,則要減1,否則不減。具體的方程見代碼。
注意細節:枚舉F的s1~s5下標時,都要N(顏色總數)開始枚舉,因為過程中某些s值會增加;
本題的啟示就是,在設計狀壓DP的時候,如果正著來不行,可以反著來,或許就能設計出符合要求的解法。
代碼:
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
using namespace std;
#define re(i, n) for (int i=0; i<n; i++)
#define re1(i, n) for (int i=1; i<=n; i++)
#define re2(i, l, r) for (int i=l; i<r; i++)
#define re3(i, l, r) for (int i=l; i<=r; i++)
#define rre(i, n) for (int i=n-1; i>=0; i--)
#define rre1(i, n) for (int i=n; i>0; i--)
#define rre2(i, r, l) for (int i=r-1; i>=l; i--)
#define rre3(i, r, l) for (int i=r; i>=l; i--)
#define ll long long
const int n = 5, MAXM = 16, MOD = 1000000007;
int m, S[n + 1];
ll F[MAXM][MAXM][MAXM][MAXM][MAXM][n], res;
void init()
{
scanf("%d", &m); int x;
re(i, m) {scanf("%d", &x); S[x]++;}
}
void solve()
{
F[S[1]][S[2]][S[3]][S[4]][S[5]][0] = 1; int tmp;
rre3(i5, m, 0) rre3(i4, m, 0) rre3(i3, m, 0) rre3(i2, m, 0) rre3(i1, m, 0) re(v, n)
if (F[i1][i2][i3][i4][i5][v]) {
if (i1) {
if (v == 1) tmp = i1 - 1; else tmp = i1;
if (tmp) {
F[i1 - 1][i2][i3][i4][i5][0] += tmp * F[i1][i2][i3][i4][i5][v];
F[i1 - 1][i2][i3][i4][i5][0] %= MOD;
}
}
if (i2) {
if (v == 2) tmp = i2 - 1; else tmp = i2;
if (tmp) {
F[i1 + 1][i2 - 1][i3][i4][i5][1] += tmp * F[i1][i2][i3][i4][i5][v];
F[i1 + 1][i2 - 1][i3][i4][i5][1] %= MOD;
}
}
if (i3) {
if (v == 3) tmp = i3 - 1; else tmp = i3;
if (tmp) {
F[i1][i2 + 1][i3 - 1][i4][i5][2] += tmp * F[i1][i2][i3][i4][i5][v];
F[i1][i2 + 1][i3 - 1][i4][i5][2] %= MOD;
}
}
if (i4) {
if (v == 4) tmp = i4 - 1; else tmp = i4;
if (tmp) {
F[i1][i2][i3 + 1][i4 - 1][i5][3] += tmp * F[i1][i2][i3][i4][i5][v];
F[i1][i2][i3 + 1][i4 - 1][i5][3] %= MOD;
}
}
if (i5) {
F[i1][i2][i3][i4 + 1][i5 - 1][4] += i5 * F[i1][i2][i3][i4][i5][v];
F[i1][i2][i3][i4 + 1][i5 - 1][4] %= MOD;
}
}
res = F[0][0][0][0][0][0];
}
void pri()
{
cout << res << endl;
}
int main()
{
init();
solve();
pri();
return 0;
}