1. 程式人生 > >hdu 1521 排列組合【指數型母函式】

hdu 1521 排列組合【指數型母函式】

題目連結:http://acm.hdu.edu.cn/showproblem.php?pid=1521

 

指數型母函式定義及其作用:

#include<stdio.h>
#include<string.h>
#include<string>
#include<queue>
#include<stack>
#include<map>
#include<vector>
#include<set>
#include<algorithm>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int N=1e5+10;

double c1[N],c2[N];

int fun(int n)
{
    int s=1;
    for(int i=1;i<=n;++i)
        s*=i;
    return s;
}

int main()
{
    int n,m;
    while(~scanf("%d %d",&n,&m))
    {
        memset(c1,0,sizeof(c1));
        memset(c2,0,sizeof(c2));
        int num[N]={0},sum[N]={0};
        for(int i=1;i<=n;++i)
        {
            scanf("%d",&num[i]);
            sum[i]=sum[i-1]+num[i];
        }
        for(int i=0;i<=num[1];i++)
            c1[i]=1.0/fun(i);
        for(int i=2;i<=n;i++)
        {
            for(int j=0;j<=sum[i-1];++j)
            {
                for(int k=0;k<=num[i];k++)
                    c2[k+j]+=c1[j]/fun(k);
            }
            for(int j=0;j<=sum[i];j++)
            {
                c1[j]=c2[j];
                c2[j]=0;
            }
        }
        printf("%.0lf\n",c1[m]*fun(m));
    }
    return 0;
}