[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题目，然后开始考虑dp状态的设定。

对于每个位置的选择，我们需要知道：一，到当前位置时颜色的剩余状态。二，前一个位置涂得颜色。

颜色的剩余状态假如强行记录的话，共有15种颜色，每种颜色有5次填涂，状态数总共是5^15。

我们发现，并不一定确切的知道每一种颜色的剩余状态，可以通过将颜色归类，来减少状态数，这里需要保证的是同类颜色的处理和对答案的影响完全一致。

可以通过颜色的剩余填涂次数，将所有颜色分为5类，分别对应剩余1次、2次、3次、4次、5次。每一类的容量最大为15，状态数为15^5。

状态转移过程不再分析。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int mod = 1e9+7;
ll dp[16][16][16][16][16][6];
int num[6];
ll dfs(int a,int b,int c,int d,int e,int prev) {    //这里写的过于恶心了，应该可以简化一下代码
    if(a==-1||b==-1||c==-1||d==-1||e==-1) return 0;
    if(dp[a][b][c][d][e][prev]!=-1) return dp[a][b][c][d][e][prev];
    ll ans = 0;
    if(a+b+c+d+e==0) ans = 1;
    else {
        int w[6];
        w[1] = a;
        w[2] = b;
        w[3] = c;
        w[4] = d;
        w[5] = e;
        if(prev>0)w[prev-1]--;
        ans += w[1]*dfs(a-1,b,c,d,e,1);
        ans %= mod;
        ans += w[2]*dfs(a+1,b-1,c,d,e,2);
        ans %= mod;
        ans += w[3]*dfs(a,b+1,c-1,d,e,3);
        ans %= mod;
        ans += w[4]*dfs(a,b,c+1,d-1,e,4);
        ans %= mod;
        ans += w[5]*dfs(a,b,c,d+1,e-1,5);
        ans %= mod;
    }
    dp[a][b][c][d][e][prev] = ans;
    return ans;
}
int main() {
    freopen("a.txt","r",stdin);
    ios::sync_with_stdio(0);
    int n;
    cin>>n;
    for(int i = 1; i <= n; i++) {
        int x;
        cin>>x;
        num[x]++;
    }
    memset(dp,-1,sizeof(dp));
    cout<<dfs(num[1],num[2],num[3],num[4],num[5],0);
    return 0;
}