HDU - 1521排列组合 指数型母函数

题目链接；
https://vjudge.net/problem/HDU-1521

题目；

有n种物品，并且知道每种物品的数量。要求从中选出m件物品的排列数。例如有两种物品A,B，并且数量都是1，从中选2件物品，则排列有"AB","BA"两种。 

Input

每组输入数据有两行，第一行是二个数n,m(1<=m,n<=10)，表示物品数，第二行有n个数，分别表示这n件物品的数量。

Output

对应每组数据输出排列数。(任何运算不会超出2^31的范围)

Sample Input

2 2
1 1

Sample Output

2

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=15;
double a[maxn],b[maxn];
int cnt[maxn];
double fac[maxn];
int n,m;
int main(){
    fac[0]=1.0;
    for(int i=1;i<=10;i++) fac[i]=i*fac[i-1];
    while(scanf("%d%d",&n,&m)!=EOF){
        memset(a,0,sizeof a);
        memset(b,0,sizeof b);
        for(int i=0;i<n;i++) cin>>cnt[i];
        for(int i=0;i<=cnt[0];i++) a[i]=1.0/fac[i];
        for(int i=1;i<n;i++){
            for(int j=0;j<=m;j++){
                for(int k=0;k<=cnt[i]&&j+k<=m;k++){
                    b[j+k]+=a[j]/fac[k];
                }
            }
            for(int j=0;j<=m;j++){
                a[j]=b[j];b[j]=0;
            }
        }
        printf("%.0f\n",a[m]*fac[m]);
    }
    return 0;
}