hdu 1521 排列组合 (指数型母函数模板题)

http://acm.hdu.edu.cn/showproblem.php?pid=1521

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

思路:指数型母函数。 对于相同的元素,需要去掉该元素的重复度,即为元素个数的阶乘。具体做法是用double类型存储(方案数除以重复度),然后在最后把阶乘乘回来四舍五入取整(为什么是四舍五入?)

 1/* ***********************************************
 2Author :111qqz
 3Created Time :2016年02月27日 星期六 19时48分31秒
 4File Name :code/hdu/1521.cpp
 5************************************************ */
 6
 7#include <cstdio>
 8#include <cstring>
 9#include <iostream>
10#include <algorithm>
11#include <vector>
12#include <queue>
13#include <set>
14#include <map>
15#include <string>
16#include <cmath>
17#include <cstdlib>
18#include <ctime>
19#define fst first
20#define sec second
21#define lson l,m,rt<<1
22#define rson m+1,r,rt<<1|1
23#define ms(a,x) memset(a,x,sizeof(a))
24typedef long long LL;
25#define pi pair < int ,int >
26#define MP make_pair
27
28using namespace std;
29const double eps = 1E-8;
30const int dx4[4]={1,0,0,-1};
31const int dy4[4]={0,-1,1,0};
32const int inf = 0x3f3f3f3f;
33double a[15],tmp[15],f[15];
34int n,m;
35int num[15];
36void pre()
37{
38    f[0] = 1.0;
39    for ( int i = 1 ;i  <= 11 ; i++)
40    {
41	f[i] = f[i-1]*i*1.0;
42    }
43}
44int main()
45{
46	#ifndef  ONLINE_JUDGE 
47	freopen("code/in.txt","r",stdin);
48  #endif
49
50	pre();
51	while (scanf("%d %d",&n,&m)!=EOF)
52	{
53	    for ( int i = 1 ; i <= n ; i++) scanf("%d",&num[i]);
54
55	    ms(a,0);
56	    ms(tmp,0);
57
58	    for ( int i = 0 ; i <= num[1] ; i++)
59	    {
60		a[i] = 1.0/f[i];
61	    }
62
63	    for ( int i = 2 ; i <= n ; i++)
64	    {
65		for ( int j = 0 ; j <= m ; j++)
66		{
67		    for ( int k = 0 ; k+j<=m&&k<=num[i] ; k++)
68		    {
69			tmp[j+k] += a[j]/f[k];
70		    }
71		}
72
73		for ( int j  = 0 ; j <= m ; j++)
74		{
75		    a[j] = tmp[j];
76		    tmp[j] = 0.0;
77		}
78	    }
79
80	    double ans = a[m]*f[m];
81	    printf("%d\n",int(ans+0.5));
82	}
83
84  #ifndef ONLINE_JUDGE  
85  fclose(stdin);
86  #endif
87    return 0;
88}