【超长篇！一篇文章讲懂生成函数！】【那些你可能忘记了的算法5】生成函数&例题&卷积-CSDN博客

1-G35 指数生成函数

#include<bits/stdc++.h>
using namespace std;
int n,m,a[11];
double fac[11],C[11],ans[11];
void calc(int m)
{
	fac[0]=1.0;for(int i=1;i<=m;i++) fac[i]=fac[i-1]*i;
	memset(C,0,sizeof(C));for(int i=0;i<=a[1];i++)C[i]=1.0/fac[i];
	for(int k=2;k<=n;k++)
	{
		memset(ans,0,sizeof(ans));
		for(int i=0;i<=m;i++)
		  for(int j=0;j<=a[k] && i+j<=m ;j++)
		     ans[i+j]+=C[i]/fac[j];
		memcpy(C,ans,sizeof(ans));
	}
	printf("%.0lf\n",C[m]*fac[m]);
}
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		for(int i=1;i<=n;i++) scanf("%d",&a[i]);
		calc(m);
	}
	return 0;
}