1079: [SCOI2008]着色方案

题目链接

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

题解：大力暴力！当然过不了
当状态过于复杂时，一般考虑简化状态，其中最常用的就是合并相同的状态
考虑到两种都能涂k次的油漆本质相同，用$f[a][b][c][d][e][l]$表示能涂1次的a个，能涂两次的b个…能涂5次的e个，前一个颜色为l的最优解
记忆化来避免访问一些无效状态

我的收获：状态的表示

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

#define P 1000000007

int n,a[7];
bool mark[16][16][16][16][16][7];
long long f[16][16][16][16][16][7];

long long dfs(int n1,int n2,int n3,int n4,int n5,int suc){
    if(n1+n2+n3+n4+n5==0) return 1;
    if(mark[n1][n2][n3][n4][n5][suc]) return f[n1][n2][n3][n4][n5][suc];
    mark[n1][n2][n3][n4][n5][suc]=1;
    long long ret=0;
    if(n1) ret+=(n1-(suc==2))*dfs(n1-1,n2,n3,n4,n5,1);
    if(n2) ret+=(n2-(suc==3))*dfs(n1+1,n2-1,n3,n4,n5,2);
    if(n3) ret+=(n3-(suc==4))*dfs(n1,n2+1,n3-1,n4,n5,3);
    if(n4) ret+=(n4-(suc==5))*dfs(n1,n2,n3+1,n4-1,n5,4);
    if(n5) ret+=n5*dfs(n1,n2,n3,n4+1,n5-1,5);
    return f[n1][n2][n3][n4][n5][suc]=ret%P;
}

void init()
{
    scanf("%d",&n);
    for(int x,i=1;i<=n;i++) cin>>x,a[x]++;
    printf("%lld",dfs(a[1],a[2],a[3],a[4],a[5],0));
}

int main()
{
    init();
    return 0;
}