Famous Stone Collector HDU - 4248 组合数学+dp

题意:给你n种物品的数量,相同物品不能区分,不同物品可以区分,问你排列的总数量。

之前做过dp的计数问题,是计算组合的总数量。排列的总数量就是状态方程改了一下。

dp[i][j]=(dp[i][j]+dp[i-1][j-k]*c[j][k]%inf)%inf;//c[i][j]:在j个位置上取k个的方案数。
因为数组大小,WA了几发。。为什么不出RE。。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;
const int maxn = 105;
int a[maxn];
long long dp[maxn][maxn*maxn];
long long c[maxn*maxn][maxn];
#define inf  1000000007

void init()
{
    c[1][0]=c[1][1]=c[0][0]=1;
    for(int i=2;i<=10010;i++)
    {
        c[i][0]=c[i][i]=1;
        for(int j=1;j<=100;j++)
            c[i][j]=(c[i-1][j]+c[i-1][j-1])%inf;
    }
    return ;
}

void Read(int &x)
{
    x=0;char c=getchar();
    bool flag=0;
    while(c<'0'||'9'<c){
        if(c=='-') flag=1;
        c=getchar();
    }
    while('0'<=c&&c<='9') {
        x=x*10+c-'0';c=getchar();
    }
    if(flag) x=-x;
}
int main()
{
    int n;
    init();
    int case1=0;
    while(~scanf("%d",&n))
    {
        int sum=0;
        for(int i=0;i<n;i++){
            Read(a[i]);
            sum+=a[i];
        }
        memset(dp,0,sizeof(dp));
        for(int i=0;i<=a[0];i++) dp[0][i]=1;
        for(int i=1;i<n;i++)
        {
            for(int j=0;j<=sum;j++)
            {
                int lim=min(j,a[i]);
                for(int k=0;k<=lim;k++)
                {
                    dp[i][j]=(dp[i][j]+dp[i-1][j-k]*c[j][k]%inf)%inf;
                }
            }
        }
        long long ans=0;
        printf("Case %d: ",++case1);
        for(int i=1;i<=sum;i++)
            ans=(ans+dp[n-1][i])%inf;
        printf("%I64d\n",ans);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值