hdu 1521 指数型母函数

题目链接：This is the link

Problem 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

Author

xhd

Recommend

xhd   |   We have carefully selected several similar problems for you:  1085 1171 1398 2082 2152 

没有什么说的，标准的模板题目

This is the code

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long LL;
double a[15],b[15],num[15];
double fac[15];
void FAC()//预处理阶乘
{
    fac[0]=1.0;
    for(int i=1; i<=10; i++)
        fac[i]=fac[i-1]*i;
}
int main()
{
    FAC();
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1; i<=n; i++)
            scanf("%lf",&num[i]);
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(int i=0; i<=num[1]; i++)
            a[i]=1.0/fac[i];
        for(int i=2; i<=n; i++)
        {
            for(int j=0; j<=m; j++)
            {
                for(int k=0; k<=num[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("%.0lf\n",a[m]*fac[m]);
    }
    return 0;
}