打这道题的时候总感觉自己在打表。。。。无话可说
用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;
}