排列组合
Time Limit : 2000/1000ms (Java/Other) Memory Limit : 65536/32768K (Java/Other)
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
题解:指数型母函数的题。
最后排列数的pr=x的r次方前的系数*r!
代码如下
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
int n, m;
int num[11];
double a[11], b[11];
int JC(int u)
{
int ans=1;
for(int i=1; i<=u; i++)
ans*=i;
return ans;
}
int main()
{
while(~scanf("%d%d", &n, &m)){
memset(num, 0, sizeof(num));
memset(a, 0, sizeof(a));
memset(b, 0, sizeof(b));
for(int i=1; i<=n; i++)
scanf("%d", &num[i]);
for(int i=0; i<=num[1]; i++){
a[i]=1.0/JC(i);
}
for(int i=2; i<=n; i++){ //一共有n项
for(int j=0; j<=m; j++){ //到m项就结束
for(int k=0; k<=num[i]; k++){
if(j+k>m)
break;
b[j+k]+=a[j]/JC(k);
}
}
for(int j=0; j<=m; j++)
{
a[j]=b[j];
b[j]=0;
}
}
printf("%.0f\n", a[m]*JC(m));
}
return 0;
}