【bzoj 1079】[SCOI2008]着色方案 逆向定义

关于dp的定义嘛一半套路一半脑洞，想到5^16的定义，不知道怎么优化结果又是套路。

5^16很大但是16^5小啊，而且只要每一种颜色出现的次数一定，他到底是哪一种 颜色并不影响所以定义f[a][b][c][d][e][x]分别表示出现了一次的颜色个数2次的。。。。但是答案还要求相邻的两个之间不允许出现同样的颜色，那么我们加一维记录上一次选用的是出现了几次的例如说上一次选用的是出现了3次的那么在处理当前出现两次的时候后发现有一种颜色是由上一次的3减一得到所以计算的时候把这一部分剔除就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#define Mod 1000000007
#define add(a,b) (a=((a)+(b))%Mod)
#define LL long long
using namespace std;
int c[6],k;
LL f[16][16][16][16][16][7];

LL dfs(int a,int b,int c,int d,int e,int x){
	LL& X=f[a][b][c][d][e][x];
	if(a+b+c+d+e==0)return X=1;
	if(X)return X;
	if(a)add(X,(LL)(a-(x==2))*dfs(a-1,b,c,d,e,1));
	if(b)add(X,(LL)(b-(x==3))*dfs(a+1,b-1,c,d,e,2));
	if(c)add(X,(LL)(c-(x==4))*dfs(a,b+1,c-1,d,e,3));
	if(d)add(X,(LL)(d-(x==5))*dfs(a,b,c+1,d-1,e,4));
	if(e)add(X,(LL)e*dfs(a,b,c,d+1,e-1,5));
	return X;
}

int main(){
	scanf("%d",&k);
	for(int x,i=1;i<=k;i++)scanf("%d",&x),c[x]++;
	printf("%d",dfs(c[1],c[2],c[3],c[4],c[5],0));
	return 0;
}