HDU - 1521 排列组合(指数型母函数)

有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

#include<iostream>  
#include<cstdio>  
#include<cstring>  
using namespace std;  
typedef long long LL;  
double a[15],b[15],num[15];  
   
double jiecheng(int n)  
{  
    double ans=1.0;  
    for(int i=1;i<=n;i++)  
        ans*=i;  
    return ans;  
}  
   
int main()  
{  
    int n,m;  
    while(cin>>n>>m){  
    for(int i=1;i<=n;i++)  
        cin>>num[i];  
    memset(a,0,sizeof(a));  
    memset(b,0,sizeof(b));  
    //a[0]=1.0;  
    for(int i=0;i<=num[1];i++)  
        a[i]=1.0/jiecheng(i);  
    for(int i=2;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]/jiecheng(k);  
            }  
        }  
        for(int j=0;j<=m;j++)  
        {  
            a[j]=b[j];  
            b[j]=0;  
        }  
    }  
    printf("%.0lf\n",a[m]*jiecheng(m));  
    }  
       
}  



猜你喜欢

转载自blog.csdn.net/dsaghjkye/article/details/79980730
今日推荐