Explosion Exploit 题解

传送门:http://codeforces.com/gym/101933/problem/E

题面

奥术飞弹!

自己和对方场地分别有 $n,m$ 只随从($1 \le n,m \le 5$),已知每个随从的血量($1 \le hp \le 6$)。

求 $d$ 个奥术飞弹将对面场地清空的概率。

分析

概率状压 $dp$ 。

显然要状压 $dp$ 一下,但是状态数有 $7^{10}$ 好像不大可做?

但是注意到有类似于 $2,1,4$ 和 $2,4,1$ 之类的重复状态,因此对于所有状态压成一个字典序最小的 5 位数。

跑一下 dp 转移即可。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <utility>
#include <map>
#define ms(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int maxn = 1000 + 5;

int change(vector<int> v){
sort(v.begin(), v.end());
int ans = 0;
for (int x : v) ans = ans * 10 + x;
return ans;
}
vector<int> change(int x){
vector<int> ans;
while (x){
if (x % 10) ans.push_back(x % 10);
x /= 10;
}
sort(ans.begin(), ans.end());
return ans;
}

int n, m, d;
vector<int> a, b;
map<PII,double> mp;

int main(){
scanf("%d%d%d", &n, &m, &d); int s = 0;
for (int i = 1, x; i <= n; i++) scanf("%d", &x), a.push_back(x), s += x;
for (int i = 1, x; i <= m; i++) scanf("%d", &x), b.push_back(x), s += x;
if (d >= s) return puts("1.00000000"), 0;
mp[make_pair(change(a), change(b))] = 1.0;
for (int i = 0; i < d; i++){
map<PII,double> mp2;
for (auto& t : mp){
vector<int> x = change(t.first.first), y = change(t.first.second);
double cnt = (int)x.size() + (int)y.size();
for (int i = 0; i < x.size(); i++){
x[i]--;
mp2[make_pair(change(x), change(y))] += t.second / cnt;
x[i]++;
}
for (int i = 0; i < y.size(); i++){
y[i]--;
mp2[make_pair(change(x), change(y))] += t.second / cnt;
y[i]++;
}
}
mp = mp2;
}
double ans = 0;
for (auto& t : mp){
if (t.first.second == 0) ans += t.second;
}
printf("%.8lf\n", ans);
return 0;
}