排列组合
Problem Description
有n种物品,并且知道每种物品的数量。要求从中选出m件物品的排列数。例如有两种物品A,B,并且数量都是1,从中选2件物品,则排列有”AB”,”BA”两种。
解题思路
本题可以转化为指数型母函数。
构造母函数:
G(x)=(10!+X1!+X22!+...+Xa1a1!)(10!+X1!+X22!+...+Xa2a2!)(10!+X1!+X22!+...+Xanan!)
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=15,ji[11]={1,1,2,6,24,120,720,5040,40320,362880,3628800};
int n,m,num[maxn];
double a[maxn],b[maxn];
int main(){
freopen("exam.in","r",stdin);
freopen("exam.out","w",stdout);
while (scanf("%d%d",&n,&m)==2){
memset(a,0,sizeof(a));a[0]=1;
for (int i=1;i<=n;i++) scanf("%d",&num[i]);
for (int i=1;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]/ji[k];
for (int j=0;j<=m;j++) a[j]=b[j],b[j]=0;
}
printf("%.0lf\n",a[m]*ji[m]);
}
return 0;
}