[BZOJ 1079][SCOI2008]着色方案

Description

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

显然5^15的算法会超时，所以要“侧过来”搞，剩余涂色次数相同的颜色可以一起计算，要注意一下处理前一种涂的颜色和当前相同的方式。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long dp[16][16][16][16][16][7];
const long long mod=1000000007;
int a[6];
long long dfs(int a,int b,int c,int d,int e,int n){
	if(dp[a][b][c][d][e][n]!=-1)return dp[a][b][c][d][e][n];
	if(a+b+c+d+e==0)return dp[a][b][c][d][e][n]=1;
	long long ret=0;
	if(a)ret+=(a-(n==2))*dfs(a-1,b,c,d,e,1);
	if(b)ret+=(b-(n==3))*dfs(a+1,b-1,c,d,e,2);
	ret%=mod;
	if(c)ret+=(c-(n==4))*dfs(a,b+1,c-1,d,e,3);
	if(d)ret+=(d-(n==5))*dfs(a,b,c+1,d-1,e,4);
	ret%=mod;
	if(e)ret+=(e-(n==6))*dfs(a,b,c,d+1,e-1,5);
	ret%=mod;
	return dp[a][b][c][d][e][n]=ret;
}
int main(){
	int k;
	scanf("%d",&k);
	int tmp;
	memset(a,0,sizeof(a));
	memset(dp,-1,sizeof(dp));
	for(int i=1;i<=k;i++){
		scanf("%d",&tmp);
		a[tmp]++;
	}
	printf("%lld\n",dfs(a[1],a[2],a[3],a[4],a[5],0));
	return 0;
}