hdu1521(排列组合母函数）

Problem Description

有n种物品，并且知道每种物品的数量。要求从中选出m件物品的排列数。例如有两种物品A,B，并且数量都是1，从中选2件物品，则排列有"AB","BA"两种。

Input

每组输入数据有两行，第一行是二个数n,m(1<=m,n<=10)，表示物品数，第二行有n个数，分别表示这n件物品的数量。

 

Output

对应每组数据输出排列数。(任何运算不会超出2^31的范围)

大概是指数型母函数的模板

能看懂吗。。

当模板记下代码就好。。

#include <iostream>

#include<stdio.h>

#include<memory.h>

using namespace std;

int f(int n)

{

    int i,s=1;

    for(i=1; i<=n; i++)

        s=s*i;

    return s;

}

int main()

{

    int n,m,i,j,k,num[12];

    double c1[12],c2[12];//因为用到了除法，所以用double型来存储。

    while(cin>>n>>m)

    {

        memset(c1,0,sizeof(c1));

        memset(c2,0,sizeof(c2));

        for(i=0; i<n; i++)

        {

            cin>>num[i];

        }

        for(i=0; i<=num[0]; i++)//赋初值

            c1[i]=1.0/f(i);

        for(i=1; i<n; i++)//n个括号

        {

            for(j=0; j<=n; j++)

                for(k=0; k<=num[i]&&k+j<=n; k++)

                    c2[k+j]+=c1[j]/f(k);

            for(j=0; j<=n; j++)

            {

                c1[j]=c2[j];

                c2[j]=0;

            }

        }

        printf("%.lf\n",c1[m]*f(m));

    }

    return 0;

}