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