乍一看母函数,后来发现复杂度太大了数组也开不了嘤嘤嘤忧桑。
查了下题解发现是dp(感觉组合数学和dp傻傻分不清楚)
接下来将下题解:
dp【i】【j】表示将i种石头分为j个石头的排列数。
新增加一种石头,假设从中间选择k个石头。之前石头数目为j个,排列为dp【i】【j】,dp【i+1】【j+k】等于把k个石头加入j个石头的的排列数,为dp【i】【j】*c【j+k】【k】
如果不选那么则为dp【i】【j】
所以dp[i+1][j+k]=dp[i+1][j+k]+dp[i][j]*c[j+k][k]为递推式。
dp[i][j]=dp[i][j]+dp[i-1][j-k]*c[j][k]
组合数
进行预处理利用c【n】【k】=c【n-1】【k】+c【n-1】【k-1】
void pre(){
c[0][0]=1;
for(int i=1;i<=N;i++)
for(int j=0;j<=110;j++)
{
if(j==0)c[i][j]=c[i-1][j];
else c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
}
}
记得mod还有一些范围。
以下为ac代码
#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N=1e4+10;
const ll mod=1e9+7;
int num;
ll dp[110][N];
ll c[N][110];
void pre(){
c[0][0]=1;
for(int i=1;i<=N;i++)
for(int j=0;j<=110;j++)
{
if(j==0)c[i][j]=c[i-1][j];
else c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
}
}
int main()
{
int n;pre();
int count=0;
while(cin>>n)
{
cout<<"Case "<<++count<<": ";
int unitnumber;
int sum=0;
memset(dp,0,sizeof(dp));
dp[0][0]=1;
for(int i=1;i<=n;i++)
{
cin>>unitnumber;
sum+=unitnumber;
for(int k=0;k<=unitnumber;k++)
for(int j=k;j<=sum;j++)
dp[i][j]=(dp[i][j]+(dp[i-1][j-k]*c[j][k]%mod))%mod;
}
ll ans=0;
for(int i=1;i<=sum;i++)
ans=(ans+dp[n][i])%mod;
cout<<ans<<endl;
}
}