大视野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

解法

神奇的动态规划，设一共有n种颜色,sum[i]表示sigma(c[j],1<=j<=i),dp[i][j]表示用了前i种颜色排成sum[i]个，其中有j个相邻同色的方案数，如果用前i个的情况去推i+1的情况，就是往sum[i]个中加入c[i+1]个，设把c[i+1]个分成k个同色的，而且不相邻地插入到sum[i]个中，设k个中有h个插入到sum[i]中相邻同色的地方，则k-h个插入到相邻不同色的地方，C[i][j]表示在i中取j个的组合数。

那么，这样插入后会有多少个相邻同色的呢，分成k组后会增加c[i+1]-k个，之后有会减少h个，所以是j+(c[i+1]-k)-h个。

然后把c[i+1]个分成k个有C[c[i+1]-1][k-1]种分法，把k组插入到sum[i+1]有C[j][h]*C[sum[i]+1-j][k-h]种方法。

所以动规方程是 dp[i+1][j+(c[i+1]-k)-h]+=C[c[i+1]-1][k-1]*dp[i][j]*C[j][h]*C[sum[i]+1-j][k-h]，最后答案是dp[n][0]。

代码

#include<cstdio>
#include<cstring>
#define LL long long
#define mod 1000000007
LL dp[20][80];
int sum[20];
int c[20];
LL C[85][85];
void pre()
{
	int i,j;
	for(i=0;i<=80;i++)
	{
		C[0][i]=0;
		C[i][0]=1;
	}
	for(i=1;i<=80;i++)
		for(j=1;j<=80;j++)
			C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
}
int main()
{
	int n,i,j,k,h;
	pre();
	scanf("%d",&n);
	sum[0]=0;
	for(i=1;i<=n;i++)
	{
		scanf("%d",&c[i]);
		sum[i]=sum[i-1]+c[i];
	}
	dp[1][c[1]-1]=1;
	for(i=1;i<n;i++)
		for(j=0;j<=sum[i]+1;j++)
			for(k=1;k<=c[i+1];k++)
				for(h=0;h<=j&&h<=k;h++)
					dp[i+1][j+(c[i+1]-k)-h]=(dp[i+1][j+(c[i+1]-k)-h]+C[c[i+1]-1][k-1]*dp[i][j]%mod*C[j][h]%mod*C[sum[i]+1-j][k-h]%mod)%mod;
	printf("%lld\n",dp[n][0]);
	return 0;
}