gym 101933 E. Explosion Exploit（概率，记忆化搜索 +剪枝）

题目链接：https://codeforces.com/gym/101933/problem/E

题目大意：有2个玩家 a 和 b，a 有 n 个棋，b 有 m 个棋，a 可以发动闪电，随机命中场上还剩下的棋子，并使得它的生命中减一，当棋子的生命值降为0时就会被抬走，a 一共可以发动 d 次闪电，求 a 能把 b 所有棋子抬走的概率（即使自己的棋子全没了也可以继续发动闪电）。

---

题解：$n,m$ 只有5，且每个棋子的生命值最大值为6，可以对当前局面进行状压，用 $dfs$ 搜索答案，可以加一些简单的剪枝例如绝对无解的情况，绝对有解的情况以及 b 已经被抬走。

这样T在第三个样例。

进一步看，当前局面与棋子的顺序无关，换一种状压方式：对于 a,b 所剩的棋子开一个桶来维护，每个桶内放对应血量的棋子，然后用12位十进制，每一位代表一个血量桶，前6位维护 b 对应血量所剩棋子数，后6位维护 a 的，再加上那些剪枝就能过。

容易发现这样状压减少了非常多状态，因为棋子的顺序被忽略了，对于(5,5,5,5,5) 这种局面，后一种状压方式只有 1种状态，而前一种有$A(5,5)$ 种状态。

---

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e4 + 10;
double esp = 1e-14;
#define pii pair<int,int>
#define fir first
#define sec second
typedef long long ll;
unordered_map<ll,double> mp;
int a[10],b[10],n,m,d;
int cnt_a[10],cnt_b[10];
ll getstate() {
	ll x = 0;
	for(int i = 6; i >= 1; i--) x = x * 10 + cnt_b[i];
	for(int i = 6; i >= 1; i--) x = x * 10 + cnt_a[i];
	return x;	
}
double dfs(int rest) {
	int cnt1 = 0,cnt2 = 0,sum1 = 0,sum2 = 0;
	for(int i = 1; i <= 6; i++)
		cnt1 += cnt_a[i],sum1 += cnt_a[i] * i;
	for(int i = 1; i <= 6; i++)
		cnt2 += cnt_b[i],sum2 += cnt_b[i] * i;
	ll x = 0;
	x = getstate();
	if(mp.count(x)) return mp[x];
	if(rest >= sum1 + sum2 || cnt2 == 0) return mp[x] = 1.0;
	if(rest < sum2 || rest == 0) return mp[x] = 0;
	if(x < 1000000) return 1;
	double p = 1.0 / (cnt1 + cnt2);
	double ans = 0;
	for(int i = 1; i <= 6; i++) {
		if(cnt_b[i]) {
			cnt_b[i]--;
			cnt_b[i - 1]++;
			ans += (cnt_b[i] + 1) * p * dfs(rest - 1);
			cnt_b[i]++;
			cnt_b[i - 1]--;
		}
	}
	for(int i = 1; i <= 6; i++) {
		if(cnt_a[i]) {
			cnt_a[i]--;
			cnt_a[i - 1]++;
			ans += (cnt_a[i] + 1) * p * dfs(rest - 1);
			cnt_a[i]++;
			cnt_a[i - 1]--;
		}
	}
	return mp[x] = ans;
}
int main() {
	scanf("%d%d%d",&n,&m,&d);
	int x;
	for(int i = 1; i <= n; i++) scanf("%d",&x),cnt_a[x]++;
	for(int i = 1; i <= m; i++) scanf("%d",&x),cnt_b[x]++;
	printf("%.12lf\n",dfs(d));
	return 0;
}