1079[bzoj]: [SCOI2008]着色方案

打这道题的时候总感觉自己在打表。。。。无话可说

用F数组表示能涂1,2,3,4,5个的油漆各有几个，pre记录前一个选的。

如果两个油漆可涂次数一样，那么他们可以看做同一个性质的，记录在一起。

然后转移，具体看代码。

#include <cstdio>
#define ll long long
using namespace std;
const int p=1000000007;
int k,x;
ll f[16][16][16][16][16][6];
int cnt[7];
ll dp(int x1,int x2,int x3,int x4,int x5,int pre)
{
    ll ans=0;
    if (f[x1][x2][x3][x4][x5][pre]) return f[x1][x2][x3][x4][x5][pre];
    if (x1+x2+x3+x4+x5==0) return 1;
    if (x1) ans+=(x1-(pre==2))*(dp(x1-1,x2,x3,x4,x5,1));
    if (x2) ans+=(x2-(pre==3))*(dp(x1+1,x2-1,x3,x4,x5,2));
    if (x3) ans+=(x3-(pre==4))*(dp(x1,x2+1,x3-1,x4,x5,3));
    if (x4) ans+=(x4-(pre==5))*(dp(x1,x2,x3+1,x4-1,x5,4));
    if (x5) ans+=(x5)*(dp(x1,x2,x3,x4+1,x5-1,5));
    f[x1][x2][x3][x4][x5][pre]=ans%p;
    return ans%p;
}
int main()
{
    register int i;
    scanf("%d",&k);
    for (i=1;i<=k;i++)
    {
        scanf("%d",&x);
        cnt[x]++;
    }
    printf("%lld",dp(cnt[1],cnt[2],cnt[3],cnt[4],cnt[5],0));
    return 0;
}