指数型母函数及模板（HDOJ 1521）

1. 指数型母函数概念

2.Example

3. 模板

HDOJ 1521

Description

有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 <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
typedef long long LL;
using namespace std;
const int N = 11;
double c1[N],c2[N];     //注意类型
LL fac[N];

void cal(){
    fac[0]=1;           //0!会被用到
    for(int i=1;i<N;i++)
        fac[i]=i*fac[i-1];
}

int main(){
    cal();
    int n,r;
    while(~scanf("%d%d",&n,&r)){
        memset(c1,0,sizeof(c1));
        memset(c2,0,sizeof(c2));

        c1[0]=1;
        int num;
        for(int i=1;i<=n;i++){
            scanf("%d",&num);
            if(num==0) continue;
            for(int j=0;j<=r;j++){
                for(int k=0;k<=num&&k+j<=r;k++){
                    c2[k+j]+=c1[j]/fac[k];
                }
            }
            for(int j=0;j<=r;j++){
                c1[j]=c2[j];
                c2[j]=0;
            }
        }

        printf("%.0lf\n",c1[r]*fac[r]);
    }

	return 0;
}