排列组合
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 5948 Accepted Submission(s): 2614
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
简单指数型生成函数的应用。
对于这道题目的生成函数如下:
这个是物品数量分别为:3,2,3的指数型生成函数。那么取m个的排列数就为x^m的系数 * m!。
这道题数据量比较小,不需要FFT,我们直接暴力多项式乘法即可。
AC代码:
#include<bits/stdc++.h>
using namespace std;
const int N=15;
int n,m,num[N];
double a[N],b[N],fac[N]={1};
void get(){
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<=num[i];j++){
for(int k=0;k+j<=n;k++) b[k+j]+=(a[k]/fac[j]);
}
for(int j=0;j<=n;j++) a[j]=b[j],b[j]=0;
}
}
int main(){
for(int i=1;i<=10;i++) fac[i]=fac[i-1]*i;
while(cin>>n>>m){
memset(a,0,sizeof a); memset(b,0,sizeof b);
for(int i=1;i<=n;i++) cin>>num[i];
get();
printf("%.0lf\n",a[m]*fac[m]);
}
return 0;
}