排列组合
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;
}