【NOIP2013模拟】粉刷匠

题目

Description
赫克托是一个魁梧的粉刷匠，而且非常喜欢思考= =
现在，神庙里有$N$根排列成一直线的石柱，从$1$到$N$标号，长老要求用油漆将这些石柱重新粉刷一遍。赫克托有$K$桶颜色各不相同的油漆，第i桶油漆恰好可以粉刷$C_i$根石柱，并且，$C_1+C_2+C_3+\cdots +C_K=N$（即粉刷$N$根石柱正好用完所有的油漆）。长老为了刁难赫克托，要求相邻的石柱颜色不能相同。
喜欢思考的赫克托不仅没有立刻开始粉刷，反而开始琢磨一些奇怪的问题，比如，一共有多少种粉刷的方案？为了让赫克托尽快开始粉刷，请你尽快告诉他答案。

Input
第一行一个正整数$T$，表示测试数据组数
对于每一组测试数据数据：
第$1$行：一个正整数$K$；
第$2$行：$K$个正整数，表示第$i$桶油漆可以粉刷的石柱个数，$C_i$。

Output
对于每组输入数据，输出一行一个整数，表示粉刷的方案数模$1000000007$。

Sample Input
3
3
1 2 3
5
2 2 2 2 2
10
1 1 2 2 3 3 4 4 5 5

Sample Output
10
39480
85937576

Data Constraint
$30\%$ $N\le 10,T\le 5$

$50\%$ $N\le 15,T\le 5$

$80\%$ $K\le 15,C_i\le 5,T\le 500$

$100\%$ $K\le 15,C_i\le 6,T\le 2000$

分析

（为了和组合数区分，题目中的$C_i$更名为$A_i$）

计数DP，先处理前缀和$S_i=\sum _{j=1}^i{A}_j$，定义$dp[i][j]$：用前$i$种颜色，刷前$S_i$个柱子，其中有$j$（$0\le j\le S_i-1$）个柱子不合法（第$x$（$1\le x\le S_i-1$）个柱子不合法，当且仅当第$x$个柱子的颜色和第$x+1$个柱子的颜色不同）的方案数。答案为$dp[k][0]$。

刷表法：将第$i+1$种颜色分段插入到$dp[i][j]$的涂法中。
枚举$k$（$1\le k\le A_{i+1}$），表示将$i+1$种颜色被分成$k$段；
枚举$t$（$0\le t\le min(j,k)$），表示我们要用分成的$k$段中的前$t$段，处理掉$t$个不合法的柱子（剩下的$k-t$段就随便放到合法的柱子旁边）。

这样处理后$dp[i][j]$就变成了$dp[i+1][j-t+(A_{i+1}-k)]$，即处理掉了$t$个不合法的，又多了$A_{i+1}-k$个不合法的（每段的最后一个插入进去后是合法的，所以要减$k$）。

• 把第$i+1$种颜色分成$k$段的方案数很简单（隔板法）：$C_{A_i-1}^{k-1}$；
• 从$j$段中选$t$个：$C_j^t$；
• 剩下的随便放：$C_{S_i+1-j}^{k-t}$（共有$S_i+1$个空，不能再放入不合法的柱子后面，所以减$j$）。

于是转移方程：$$dp[i+1][j-t+(A_{i+1}-k)]=dp[i][j]\times C_{A_i-1}^{k-1}\times C_{S_i+1-j}^{k-t}\times C_j^t$$
边界：$$dp[0][0]=1$$
要从$i=0$开始刷表。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

#define MAXC 6
#define MAXK 15
#define MAXN MAXK*MAXC
#define MOD 1000000007
int K;
int A[MAXK+5],S[MAXK+5];
int dp[MAXK+5][MAXN+5];

int C[MAXN+5][MAXN+5];
void Init(int N){//预处理组合数
    C[0][0]=1;
    for(int i=1;i<=N;i++){
        C[i][0]=1;
        for(int j=1;j<=N;j++)
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%MOD;
    }
}

int main(){
    int T;
    scanf("%d",&T);
    Init(MAXN);
    while(T--){
        scanf("%d",&K);
        for(int i=1;i<=K;i++)
            scanf("%d",&A[i]),S[i]=S[i-1]+A[i];
        memset(dp,0,sizeof dp);
        dp[0][0]=1;
        for(int i=0;i<K;i++){//注意这里
            for(int j=0;j<=S[i];j++){
                for(int k=1;k<=A[i+1];k++){
                    for(int t=0;t<=min(k,j);t++){
                        dp[i+1][j-t+A[i+1]-k]=(dp[i+1][j-t+A[i+1]-k]+(long long)dp[i][j]*C[A[i+1]-1][k-1]%MOD*C[S[i]+1-j][k-t]%MOD*C[j][t]%MOD)%MOD;
                    }
                }
            }
        }
        printf("%d\n",dp[K][0]);
    }
}