着色方案（状压DP）

　　有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

 

以15^5状压，而不是5^15

  

#include<stdio.h>
#include<string.h>
const int mod = 1e9+7;

long long dp[20][20][20][20][20][6];//从左到右分别记录个数为１，２，３，４，５的色块的种类，上一个所涂的色块在上一次的个数
long long a[20];

long long bfs(long long a,long long b,long long c,long long d,long long e,long long last)
{
    if(dp[a][b][c][d][e][last] != -1)
        return dp[a][b][c][d][e][last];
    if(a +b+c+d+e == 0)
        return dp[a][b][c][d][e][last] =1;
    long long ans = 0;
    if(a>0)//如果色块为1的色块的种类大于０
        ans +=(a-(last==2))*bfs(a-1,b,c,d,e,1),ans%=mod;//如果上一个last＝＝２,代表上一个位置的色块在上一次的个数是２,那么放完之后，这一次个数就是１，这类色块不能再放了，但是其他类的个数为1的色块还可以放，同时，色块被放，递归a-1，
    if(b>0)
        ans += (b-(last==3))*bfs(a+1,b-1,c,d,e,2),ans%=mod;//色块个数为2的色块被放一个，个数变为１，a+1,b-1
    if(c>0)
        ans += +(c-(last==4))*bfs(a,b+1,c-1,d,e,3),ans%=mod;
    if(d>0)
        ans += (d-(last==5))*bfs(a,b,c+1,d-1,e,4),ans%=mod;
    if(e>0)
        ans += e*bfs(a,b,c,d+1,e-1,5),ans%=mod;//上一个被放的色块不可能现在的个数是5，因为５是最大值，色块被放完，会减少

    return dp[a][b][c][d][e][last] = ans%mod;

}

int main()
{
    long long i,n,num;
    memset(dp,-1,sizeof(dp));
    memset(a,0,sizeof(a));
    scanf("%lld",&n);

    for(i = 0; i < n; i ++)
    {
        scanf("%lld",&num);
        a[num]++;
    }

    num = bfs(a[1],a[2],a[3],a[4],a[5],0);//a[i]代表色块个数为i的色块的种类
    printf("%lld\n",num);
    return 0;
}