hdu 4248 A Famous Stone Collector dp+组合数学

预处理出c[10000][100],然后dp递推。

#include <iostream>
#include <cstdio>
#include <cstring>
#define ll long long
using namespace std;
const int maxn=1e4+9,mod=1e9+7;
ll dp[maxn],c[maxn][111];
int a[maxn];
int n,sum;

ll getdp()
{
    memset(dp,0,sizeof(dp));
    dp[0]=1;
    for(int k=1;k<=n;k++)
    for(int i=sum;i>=0;i--)
    for(int j=1;j<=a[k]&&i+j<=sum;j++)
    dp[i+j]=(dp[i+j]+dp[i]*c[i+j][j])%mod;
    ll ret=0;
    for(int i=1;i<=sum;i++)
    ret=(ret+dp[i])%mod;
    return ret;
}

void getc()
{
    for(int i=0;i<=1e4;i++) c[i][0]=1;

    for(int i=1;i<=1e4;i++)
    for(int j=1;j<=100&&j<=i;j++)
    {
        if(i==j) c[i][j]=1;
        else c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod;
    }
}

int main()
{
//    freopen("in.txt","r",stdin);
    int tt=0;
    getc();
    while(scanf("%d",&n)!=EOF)
    {
        sum=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            sum+=a[i];
        }
        ll ans=getdp();
        printf("Case %d: %I64d\n",++tt,ans);
    }
    return 0;
}


你可能感兴趣的:(hdu 4248 A Famous Stone Collector dp+组合数学)