HDU 4532 组合数+动态规划

最新推荐文章于 2022-11-13 13:11:19 发布

原创最新推荐文章于 2022-11-13 13:11:19 发布 · 479 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

------DP------ 同时被 2 个专栏收录

2 篇文章

订阅专栏

组合数学

2 篇文章

订阅专栏

本文介绍了如何运用动态规划策略解决HDU 4532编程题中涉及的组合数问题。通过分析题目，建立了状态转移方程，并给出详细解题步骤，帮助读者理解动态规划在解决此类问题中的应用。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=47+3;
const int maxm=447+10;
int A[maxn];
typedef long long LL;
const int mod=1e9+7;
LL fac[maxm],inv[maxm];
LL C[maxm][maxm];
int kase=0;
void init(int n)
{
    fac[0]=1;
    for(int i=1;i<=n;i++){
        fac[i]=fac[i-1]*i%mod;
    }
    inv[0]=1;
    inv[1]=1;
    for(int i=2;i<=n;i++){
        inv[i]=inv[mod%i]*(mod-mod/i)%mod;
    }
    for(int i=1;i<=n;i++){
        inv[i]=inv[i-1]*inv[i]%mod;
    }
    memset(C,0,sizeof(C));
    for(int i=0;i<=n;i++){
        for(int j=0;j<=i;j++){
            C[i][j]=fac[i]*inv[i-j]%mod*inv[j]%mod;
        }
    }
}
int T,N;
LL dp[maxn][maxm];
void solve()
{

    scanf("%d",&N);
    for(int i=1;i<=N;i++){
        scanf("%d",&A[i]);
    }
    memset(dp,0,sizeof(dp));
    dp[1][A[1]-1]=1;
    int sumv=A[1];
    for(int i=2;i<=N;i++){
        for(int j=0;j<sumv;j++){
            for(int k=1;k<=A[i];k++){
                int up=min(j,k);
                for(int p=0;p<=up;p++){
                    (dp[i][A[i]-k+j-p]+=dp[i-1][j]*C[A[i]-1][k-1]%mod*C[j][p]%mod*C[sumv+1-j][k-p]%mod)%=mod;
                }
            }
        }
        sumv+=A[i];
    }
    LL ans=dp[N][0];
    for(int i=1;i<=N;i++){
        (ans*=fac[A[i]])%=mod;
    }
    if(ans<0)   ans+=mod;
    printf("Case %d: %lld\n",kase,ans);
}

int main()
{
    init(maxm-1);
    scanf("%d",&T);
    kase=0;
    while(T--){
        kase++;
        solve();
    }

    return 0;
}