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


  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值