【bzoj 1079】 着色方案 【SCOI2008】

Description

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

Input

　　第一行为一个正整数k，第二行包含k个整数c1, c2, ... , ck。

Output

　　输出一个整数，即方案总数模1,000,000,007的结果。

Sample Input

3
1 2 3

Sample Output

10

HINT

 

 100%的数据满足：1 <= k <= 15, 1 <= ci <= 5

这道题是一个玄妙的DP，显然方案数只和可以涂次的颜色数有关(即可以涂相同次数的颜色是等价的)，表示可以涂次的颜色数为，上次涂得颜色可以涂last次的方案数，然后记忆化搜索一下就可以了，下面是程序：

#include<stdio.h>
#include<iostream>
#define ll long long
using namespace std;
const int N=16,mod=1e9+7;
ll f[N][N][N][N][N][6];
ll dp(int a,int b,int c,int d,int e,int last){
	if(f[a][b][c][d][e][last]){
		return f[a][b][c][d][e][last];
	}
	if(!(a|b|c|d|e)){
		return f[a][b][c][d][e][last]=1;
	}
	ll tp=0;
	if(a){
		tp+=(a-(last==2))*dp(a-1,b,c,d,e,1);
		tp%=mod;
	}
	if(b){
		tp+=(b-(last==3))*dp(a+1,b-1,c,d,e,2);
		tp%=mod;
	}
	if(c){
		tp+=(c-(last==4))*dp(a,b+1,c-1,d,e,3);
		tp%=mod;
	}
	if(d){
		tp+=(d-(last==5))*dp(a,b,c+1,d-1,e,4);
		tp%=mod;
	}
	if(e){
		tp+=e*dp(a,b,c,d+1,e-1,5);
		tp%=mod;
	}
	return f[a][b][c][d][e][last]=tp;
}
int main(){
	int n,i,x[6]={0},t;
	scanf("%d",&n);
	for(i=0;i<n;i++){
		scanf("%d",&t);
		x[t]++;
	}
	printf("%lld\n",dp(x[1],x[2],x[3],x[4],x[5],0));
	return 0;
}