着色方案（记忆化搜索）

传送门
有n个木块排成一行，从左到右依次编号为1~n。
你有k种颜色的油漆，其中第i种颜色的油漆足够涂ci个木块。 所有油漆刚好足够涂满所有木块，即c1+c2+…+ck=n。
相邻两个木块涂相同色显得很难看，所以你希望统计任意两个相邻木块颜色不同的着色方案。
输入描述:
第一行为一个正整数k，第二行包含k个整数c1, c2, … , ck。
输出描述:
输出一个整数，即方案总数模1,000,000,007的结果。
示例1
输入
复制
3
1 2 3
输出
复制
10
1<=k<=15 ,1<= ci <=5

思路：
首先由数据范围我们可以发现最多只有5种颜色，且每种颜色使用次数也很小，所以我们可以采用六维度数组dp[a][b][c][d][e][pre]表示，有a种可以用1次的颜色，b种可以用2次的颜色，c种可以用3次的颜色，d种可以用四次的颜色，e种可以用五次的颜色，在前一次为用可以pre次的颜色情况下符合条件的情况。

注意：必须用添加pre这维不然，情况会错误，当pre为某个的时候为0，其他情况为会计算为0；

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll dp[17][17][17][17][17][6];
ll mod=1e9+7;
int n,a[6];
ll dfs(int a,int b,int c,int d,int e,int pre){
	if(dp[a][b][c][d][e][pre]!=-1) return dp[a][b][c][d][e][pre];
	ll ans=0;
	if(a) ans=(ans+(a-(pre==2))*dfs(a-1,b,c,d,e,1))%mod;
	if(b) ans=(ans+(b-(pre==3))*dfs(a+1,b-1,c,d,e,2))%mod;
	if(c) ans=(ans+(c-(pre==4))*dfs(a,b+1,c-1,d,e,3))%mod;
	if(d) ans=(ans+(d-(pre==5))*dfs(a,b,c+1,d-1,e,4))%mod;
	if(e) ans=(ans+e*dfs(a,b,c,d+1,e-1,5))%mod;
	return dp[a][b][c][d][e][pre]=ans;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n;
	int x;
	for(int i=1;i<=n;i++){
		cin>>x;
		++a[x];
	}
	memset(dp,-1,sizeof(dp));
	for(int i=0;i<=5;i++){
		dp[0][0][0][0][0][i]=1;
	}
	cout<<dfs(a[1],a[2],a[3],a[4],a[5],0)<<endl;
}