1. 程式人生 > >GYM 101933E(記憶化搜索)

GYM 101933E(記憶化搜索)

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(記憶化搜索)