GYM 101933E(記憶化搜索)
阿新 • • 發佈:2019-05-11
dfs dep cst std using code 情況 urn main
用每個人的血量作為狀態去搜索T飛,考慮題解中更好的搜索方式:每種血量有多少個人作為狀態。這樣會減去很多重復的狀態,因為只要乘一下就得到了所有相同情況的和。
雖然我不會算,但是直觀感受起來復雜度比較優秀。
#include <cstdio> #include <unordered_map> using namespace std; typedef double db; typedef long long ll; int n, m, d, mod = 1e6; int cnt[2][10]; unordered_map<ll, db> mp; void read(int n, int k) { for (int a, i = 1; i <= n; i++) { scanf("%d", &a); cnt[k][a]++; } } ll tran(ll res = 0) { for (int i = 0; i < 2; i++) for (int j = 1; j <= 6; j++) res = res * 10 + cnt[i][j]; return res; } db dfs(ll t, int depth) { if (mp.count(t)) return mp[t]; if (t % mod == 0) return mp[t] = 1; if (depth == d) return mp[t] = 0; db res = 0; int tmp = 0; for (int i = 0; i < 2; i++) for (int j = 1; j <= 6; j++) tmp += cnt[i][j]; for (int i = 0; i < 2; i++) { for (int j = 1; j <= 6; j++) { if (cnt[i][j]) { cnt[i][j]--; cnt[i][j - 1]++; res += dfs(tran(), depth + 1) * (cnt[i][j] + 1) / tmp; cnt[i][j]++; cnt[i][j - 1]--; } } } return mp[t] = res; } int main() { scanf("%d %d %d", &n, &m, &d); read(n, 0), read(m, 1); return !printf("%.8lf\n", dfs(tran(), 0)); }
GYM 101933E(記憶化搜索)