题目链接:Hearthstone
题目大意:
牌堆有n张奥术牌,奥术牌可以再从牌堆摸两张牌, m张伤害牌,伤害各为
xi
,初始从牌堆摸一张,问本回合能击杀给定hp的对手的概率,结果用分数表示。(n+m<=20)
如果n < m, 回合中能抽取的牌数为k-1张奥术牌, k张伤害牌,
枚举所有的伤害牌组合,如果 k张伤害和>= hp且k-1<=n,说明这种牌组能被抽到并且能击杀对手,计算出这种伤害组合的所有牌序组合数
dp[k−1][k]∗Ck−1n∗(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并且 ∑xi>=hp ,多了一种特殊情况,牌堆的牌数不够抽,只能n张奥术,m张伤害(而不是n+1),加上dp[n][m] * n! * m!;
最后答案除以总牌序(m+n)!
复杂度
O(2m)
#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;
}