题目
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个木块排成一行,从左到右依次编号为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;
}