@OceanEye7年前
05/29
21:46
dp[a][b][c][d][e][pre] = dp[a-1][b][c][d][e][1]*系数 + … + dp[a][b][c][d][e-1][5]*系数
其中a, b, c, d, e是能涂一块砖的颜色数,二,三……五
pre是上一次用的颜色是能涂pre块砖头的
因为上一次用的是能涂pre块砖头的,所以这一次涂色里能涂pre-1块砖头的颜色里有一种颜色不能用了
对应的系数就要-1
code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
#include <cstdio> #include <algorithm> #include <cstring> #include <cstdlib> #include <cmath> #define ll long long #define INF 1000000000 #define MOD 1000000007 #define p(x) ('0'<=x&&x<='9') #define For(i,l,r) for(int i=l;i<=r;i++) char cc; int C; template <class T> void read( T &x ) { x=0; cc=getchar(); C=1; while(!p(cc)) { if(cc=='-') C=-1; cc=getchar(); } while(p(cc)) { x=x*10+cc-48; cc=getchar(); } x*=C; } using namespace std; ll dp[16][16][16][16][16][6]; #define NOW dp[a][b][c][d][e][pre] ll dfs(int a,int b,int c,int d,int e,int pre) { if(NOW) return NOW; if(!(a||b||c||d||e)) return NOW; if(a>0) NOW+=(a-(pre==2))*dfs(a-1,b,c,d,e,1); if(b>0) NOW+=(b-(pre==3))*dfs(a+1,b-1,c,d,e,2); if(c>0) NOW+=(c-(pre==4))*dfs(a,b+1,c-1,d,e,3); if(d>0) NOW+=(d-(pre==5))*dfs(a,b,c+1,d-1,e,4); if(e>0) NOW+=e*dfs(a,b,c,d+1,e-1,5); NOW%=MOD; return NOW; } int n,_,cnt[6]; int main() { for(int i=0;i<=5;i++) dp[0][0][0][0][0][i]=1; read(n); for(int i=1;i<=n;i++) read(_),cnt[_]++; printf("%lld",dfs(cnt[1],cnt[2],cnt[3],cnt[4],cnt[5],0)); return 0; } |