hdu 4248 A Famous Stone Collector

来源：互联网发布：死或生5mac 编辑：程序博客网时间：2024/06/05 20:06

http://acm.hdu.edu.cn/showproblem.php?pid=4248

组合数学dp：

dp[ i ][ j ] 表示用了前 i 个数组成长度为 j 的个数

#include <iostream>#include <cstdio>#include <cstring>using namespace std;const int mod=1000000007;long long dp[110][11000];long long c[11000][110],num[110];int main(){    c[0][0]=1;    c[1][0]=c[1][1]=1;    for(int i=2;i<=10000;i++)    {        c[i][0]=c[i][i]=1;        for(int j=1;j<i&&j<=100;j++)          c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod;    }    int n,sum,cas=1;    while(scanf("%d",&n)==1)    {        sum=0;        for(int i=0;i<n;i++)  scanf("%d",&num[i]),sum+=num[i];        memset(dp,0,sizeof(dp));        dp[0][0]=1;        for(int i=0;i<n;i++)        {            for(int j=0;j<=sum;j++)  dp[i+1][j]+=dp[i][j];            for(int j=0;j<=sum;j++)              if(dp[i][j])              {                  for(int r=1;r<=num[i]&&j+r<=sum;r++)  dp[i+1][j+r]=(dp[i+1][j+r]+dp[i][j]*c[j+r][r])%mod;              }        }        long long ans=0;        for(int i=1;i<=sum;i++)  ans=(ans+dp[n][i])%mod;        printf("Case %d: %I64d\n",cas++,ans);    }    return 0;}