HDU 1521 指数型母函数

 

方法一： DFS

方法二：生成函数

每个数可以重复一定次数，求排列组合数，这是裸的指数型生成函数；

#include <bits/stdc++.h>

using namespace std;

double c1[15],c2[15];
int a[15];
int num[15];


int main() {

    a[0] = 1;
    for(int i=1; i<13; i++)
        a[i] = a[i-1]*i;

    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF) {

        memset(c1,0,sizeof(c1));
        memset(c2,0,sizeof(c2));
        
        for(int i=1; i<=n; i++)
            scanf("%d",&num[i]);

        for(int i=0; i<=num[1]; i++)
            c1[i] = 1.0/a[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++)
                    c2[j+k] +=c1[j]/a[k];
            }
            memcpy(c1,c2,sizeof(c2));
            memset(c2,0,sizeof(c2));
        }

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

    }

    return 0;
}