hdu 4248 A Famous Stone Collector

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;
}

你可能感兴趣的:(hdu 4248 A Famous Stone Collector)