bzoj1079 [SCOI2008]着色方案（计数类DP）

题目

有n个木块排成一行，从左到右依次编号为1~n。你有k种颜色的油漆，其中第i种颜色的油漆足够涂ci个木块。
所有油漆刚好足够涂满所有木块，即c1+c2+...+ck=n。相邻两个木块涂相同色显得很难看，所以你希望统计任意两
个相邻木块颜色不同的着色方案。（100%的数据满足：1 <= k <= 15, 1 <= ci <= 5）

题解

计数类DP
看到ci如此小，不妨从ci下手。设f[c5][c4][c3][c2][c1][last]表示有5次使用机会的有c5个，……，1次的有c1个，最后一次使用的是剩余使用次数last的颜料。
转移时，需要注意这次使用的颜料l，下次使用时不能再使用颜料l。所以次数要减1。方案数=使用5次+使用4次……

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=20;
const int mod=1e9+7;

ll f[16][16][16][16][16][16];
bool v[16][16][16][16][16][6];
ll dfs(int a,int b,int c,int d,int e,int l)
{
    if(v[a][b][c][d][e][l]) return f[a][b][c][d][e][l];
    v[a][b][c][d][e][l]=true;
    ll re=0;
    if(a>0) re+=dfs(a-1,b+1,c,d,e,5)*a;
    if(b>0) re+=dfs(a,b-1,c+1,d,e,4)*(b-(l==5));
    if(c>0) re+=dfs(a,b,c-1,d+1,e,3)*(c-(l==4));
    if(d>0) re+=dfs(a,b,c,d-1,e+1,2)*(d-(l==3));
    if(e>0) re+=dfs(a,b,c,d,e-1,0)  *(e-(l==2));
    return f[a][b][c][d][e][l]=re%mod;
}

int s[maxn];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int col;
        scanf("%d",&col);
        s[col]++;
    }
    
//    memset(f,-1,sizeof(f));
    for(int i=0;i<=5;i++) v[0][0][0][0][0][i]=true,f[0][0][0][0][0][i]=1;
    printf("%lld\n",dfs(s[5],s[4],s[3],s[2],s[1],0));
    return 0;
}