[bzoj1079]着色方案

1 #include <iostream> 2 #include <algorithm> 3 #include <cstdio> 4 #include <cstring> 5 #include <cstdlib> 6 #include <queue> 7 #include <string> 8 #include <vector> 9 using namespace std; 10 11 int n,x,tot[10]; 12 long long dp[16][16][16][16][16][16]; 13 long long f(int a,int b,int c,int d,int e,int l){ 14 if(dp[a][b][c][d][e][l]!=-1)return dp[a][b][c][d][e][l]; 15 #define t dp[a][b][c][d][e][l] 16 t=0; 17 if(a+b+c+d+e==0)return t=1; 18 if(a>0)t+=(a-(l==2))*f(a-1,b,c,d,e,1); 19 if(b>0)t+=(b-(l==3))*f(a+1,b-1,c,d,e,2); 20 if(c>0)t+=(c-(l==4))*f(a,b+1,c-1,d,e,3); 21 if(d>0)t+=(d-(l==5))*f(a,b,c+1,d-1,e,4); 22 if(e>0)t+=e*f(a,b,c,d+1,e-1,5); 23 return t%=1000000007; 24 } 25 int main(){ 26 scanf("%d",&n); 27 for(int i=1;i<=n;i++)scanf("%d",&x),tot[x]++; 28 memset(dp,-1,sizeof(dp)); 29 printf("%lld\n",f(tot[1],tot[2],tot[3],tot[4],tot[5],0)); 30 }