bzoj1079: [SCOI2008]着色方案(DP)

1079: [SCOI2008]着色方案

题目：传送门 

 

题解：

　　 DP刚神多年前讲过的一道神题。

　　 二话不说，上来就是一个六维数组：F[i][a][b][c][d][e]//表示上一次涂的颜色是还剩下i次可用的，a~e表示不同次数的颜色种数。

　　 次数一样的颜色其实是相同的

　　 那么我们记忆化搜索一下就OK（表示很久没打过了...）

代码（巨丑勿喷）：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
const int mod=1000000007;
typedef long long LL;
LL f[6][16][16][16][16][16];//f[i][a][b][c][d][e]表示上一次涂的颜色是还剩下i次可用的，a~e表示不同次数的颜色种数 
int a[6];
int k;
LL DP(int last,int a1,int a2,int a3,int a4,int a5)
{
    if(a1<0 || a2<0 || a3<0 || a4<0 || a5<0)return 0;
    if(a1==0 && a2==0 && a3==0 && a4==0 && a5==0)return 1;
    if(f[last][a1][a2][a3][a4][a5]!=0) return f[last][a1][a2][a3][a4][a5];
    
    LL sum=0;
    sum=( sum + ( ( a1 - ( ( last==2 ) ? 1 : 0 ) ) * DP ( 1,a1-1,a2,a3,a4,a5 ) ) ) % mod;
    sum=( sum + ( ( a2 - ( ( last==3 ) ? 1 : 0 ) ) * DP ( 2,a1+1,a2-1,a3,a4,a5) )) % mod;
    sum=( sum + ( ( a3 - ( ( last==4 ) ? 1 : 0 ) ) * DP ( 3,a1,a2+1,a3-1,a4,a5) )) % mod;
    sum=( sum + ( ( a4 - ( ( last==5 ) ? 1 : 0 ) ) * DP ( 4,a1,a2,a3+1,a4-1,a5) )) % mod;
    sum=( sum + a5*DP(5,a1,a2,a3,a4+1,a5-1))%mod;
    f[last][a1][a2][a3][a4][a5]=sum;
    return sum;
}
int main()
{
    memset(f,0,sizeof(f));
    scanf("%d",&k);
    for(int i=1;i<=k;i++)
    {
        int x;
        scanf("%d",&x);
        a[x]++;
    }
    printf("%lld\n",DP(0,a[1],a[2],a[3],a[4],a[5]));
    return 0;
}