hdu 1521

最新推荐文章于 2021-02-12 22:05:40 发布

chengouxuan

最新推荐文章于 2021-02-12 22:05:40 发布

阅读量832

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM 文章标签： output input c

本文链接：https://blog.csdn.net/chengouxuan/article/details/6937637

ACM 专栏收录该内容

83 篇文章

订阅专栏

排列组合

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 811 Accepted Submission(s): 325

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

Author

xhd

Recommend

xhd

指数生成函数！黑书P. 257

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cassert>
#include <cmath>
#include <algorithm>

#define LEN 100

void poly_mult (double *a, double *b)
{
    double c[LEN];
    std::fill (c, c + LEN, 0.0);
    for (int i=0; i<LEN; ++i) {
        for (int k=0; k+i<LEN; ++k)
            c[i + k] += a[i] * b[k];
    }
    memcpy (a, c, sizeof (c));
}

#define FACS 100

double fac [FACS];

void init_fac ()
{
    *fac = 1;
    for (int i=1; i<FACS; ++i)
        fac[i] = i * fac[i - 1];
}

#define POLYS 20

void solve ()
{
    double poly[POLYS][LEN];
    for (int i=0; i<POLYS; ++i)
        std::fill (poly[i], poly[i] + LEN, 0.0);

    int n, m;
    if (scanf ("%d%d", &n, &m) == EOF)
        exit (0);

    for (int i=0; i<n; ++i) {
        int t;
        scanf ("%d", &t);
        for (int k=0; k<=t; ++k)
            poly[i][k] = 1.0 / fac[k];
    }

    double res[LEN];
    std::fill (res, res + LEN, 0.0);
    res[0] = 1.0;

    for (int i=0; i<n; ++i)
        poly_mult (res, poly[i]);

    printf ("%.0lf\n", res[m] * fac[m]);
}

int main ()
{
    init_fac ();
    while (1)
        solve ();
    return 0;
}