hdu5816 多校7 Hearthstone【组合计数+dp】

题目链接：Hearthstone

题目大意：
牌堆有n张奥术牌，奥术牌可以再从牌堆摸两张牌， m张伤害牌，伤害各为$x_i$，初始从牌堆摸一张，问本回合能击杀给定hp的对手的概率，结果用分数表示。(n+m<=20)

---

如果n < m, 回合中能抽取的牌数为k-1张奥术牌， k张伤害牌，
枚举所有的伤害牌组合，如果 k张伤害和>= hp且k-1<=n，说明这种牌组能被抽到并且能击杀对手，计算出这种伤害组合的所有牌序组合数 $dp[k-1][k] * C^{k-1}_n * （k-1）! * k！ * (n+m-2*k+1)!$，累加即可;
其中dp[i][j] 为将奥术牌看做一种，伤害牌看做一种下，i 张奥术和 j 张伤害的组合数，可以看出i>=j-1,

初始dp[0][0] = 0, dp[0][1] = 1;
递推关系为dp[i][0] = 1, dp[i][j] = dp[i-1][j] + dp[i][j-1] (j < i), dp[i][j] = dp[i][[j-1] (i<=j<=i+1);
(感觉不怎么好想，渣渣打表找的规律。。）

如果n >= m并且$\sum x_i >= hp$，多了一种特殊情况，牌堆的牌数不够抽，只能n张奥术，m张伤害（而不是n+1），加上dp[n][m] * n! * m！；

最后答案除以总牌序(m+n)!
复杂度$O(2^m)$

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long LL;
int hp, n, m;
int x[25];
LL c[25][25], a[25], dp[25][25];   //c: 组合数 a: 阶乘  dp：i奥术j伤害组合数
LL ans;

void init() //预处理
{
    c[0][0] = 1;
    for(int i = 1; i <= 20; i++)
    {
        c[i][0] = 1;
        for(int j = 1;j <= i; j++)
            c[i][j] = c[i-1][j-1] + c[i-1][j];
    }

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

    memset(dp, 0, sizeof(dp));
    dp[0][0] = 0;
    dp[0][1] = 1;
    for(int i = 1; i <= 20; i++)
    {
        dp[i][0] = 1;
        for(int j = 1; j < i; j++)
            dp[i][j] = dp[i-1][j] + dp[i][j-1];
        dp[i][i] = dp[i][i+1] = dp[i][i-1];
    }    
}

void dfs(int cur, int s, int k)
{
    if(k > n+1)
        return;
    for(int i = cur; i < m; i++)
    {
        if(s+x[i] >= hp)
        {
            if(k == 1)
                ans += a[n+m-1];
            else
                ans += dp[k-1][k] * c[n][k-1] * a[k-1] * a[k] * a[n+m-2*k+1];
            if(k == m && n >= m)
                ans += dp[n][m] * a[n] * a[m];
        }
        dfs(i+1, s+x[i], k+1);
    }
}

int main(int argc, char const *argv[])
{
    init();
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d%d", &hp, &n, &m);
        for(int i = 0; i < m; i++)
            scanf("%d", &x[i]);
        ans = 0;
        dfs(0, 0, 1);

        LL all = a[n+m];
        LL g = __gcd(ans, all);

        if(ans)
            printf("%I64d/%I64d\n", ans/g, all/g);
        else
            printf("0/1\n");
    }
    return 0;
}