hdu 1521 排列組合【指數型母函式】
阿新 • • 發佈:2018-12-07
題目連結:http://acm.hdu.edu.cn/showproblem.php?pid=1521
指數型母函式定義及其作用:
#include<stdio.h> #include<string.h> #include<string> #include<queue> #include<stack> #include<map> #include<vector> #include<set> #include<algorithm> #define inf 0x3f3f3f3f using namespace std; typedef long long ll; const int N=1e5+10; double c1[N],c2[N]; int fun(int n) { int s=1; for(int i=1;i<=n;++i) s*=i; return s; } int main() { int n,m; while(~scanf("%d %d",&n,&m)) { memset(c1,0,sizeof(c1)); memset(c2,0,sizeof(c2)); int num[N]={0},sum[N]={0}; for(int i=1;i<=n;++i) { scanf("%d",&num[i]); sum[i]=sum[i-1]+num[i]; } for(int i=0;i<=num[1];i++) c1[i]=1.0/fun(i); for(int i=2;i<=n;i++) { for(int j=0;j<=sum[i-1];++j) { for(int k=0;k<=num[i];k++) c2[k+j]+=c1[j]/fun(k); } for(int j=0;j<=sum[i];j++) { c1[j]=c2[j]; c2[j]=0; } } printf("%.0lf\n",c1[m]*fun(m)); } return 0; }