HDU 1521 排列组合（指数型母函数）

HDU 1521 排列组合（指数型母函数）：http://acm.hdu.edu.cn/showproblem.php?pid=1521

题面描述：

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)

Total Submission(s): 3369    Accepted Submission(s): 1409


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

题目大意：

已知有a1种1物品，a2种2物品，.... ，an种n物品，从中任意取m个，求所有能取到排列的方法数。

题目分析：

一开始想到的是用全排列公式来求解此题，但是只有当m==n时，可以用全排列公式，当m<n时，也有可以用的公式，即为n^m种，而当m>n时，就没有公式了。后来又根据此题的有序性，想到了指数型母函数，其实也就是一个指数型母函数的模板题。

先计算出全排列公式，然后取出m个物品的全排列数就是第m项的系数。

代码实现：

#include <iostream>
#include <cstdio>
#include <math.h>
#include <cstring>
using namespace std;

int a[20];
double jc[20];
double a1[20],a2[20];

int main()
{
int n,m;
long long sum;
jc[0]=1;
for(int i=1;i<=10;i++)
{
jc[i]=i*jc[i-1];
}
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
}
memset(a1,0,sizeof(a1));
memset(a2,0,sizeof(a2));

for(int i=0;i<=a[0];i++)
{
a1[i]=1.0/jc[i];
}
for(int i=1;i<n;i++)
{
for(int j=0;j<=m;j++)
{
for(int k=0;k<=a[i]&&k+j<=m;k++)
{
a2[j+k]+=(a1[j]*1.0/jc[k]);
}
}
for(int j=0;j<=m;j++)
{
a1[j]=a2[j];
a2[j]=0;
}
}
printf("%.0lf\n",a1[m]*jc[m]);
}
return 0;
}