結果

問題 No.58 イカサマなサイコロ
ユーザー oooooba
提出日時 2021-09-23 21:00:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,174 ms / 5,000 ms
コード長 1,249 bytes
コンパイル時間 1,314 ms
コンパイル使用メモリ 117,836 KB
最終ジャッジ日時 2025-01-24 16:20:33
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <array>
#include <cmath>
#include <cstdio>
#include <deque>
#include <iomanip>
#include <iostream>
#include <numeric>
#include <optional>
#include <queue>
#include <set>
#include <stack>
#include <unordered_map>
#include <unordered_set>
#include <vector>

using namespace std; // NOLINT

void dfs(int32_t sum, int32_t c, unordered_map<int32_t, int32_t> &hist) {
    if (c == 0) {
        ++hist[sum];
        return;
    }
    for (auto i = 1; i <= 6; ++i) {
        dfs(sum + i, c - 1, hist);
    }
}

void dfs2(int32_t sum, int32_t c, int32_t d,
          unordered_map<int32_t, int32_t> &hist) {
    if (c == 0) {
        dfs(sum, d, hist);
        return;
    }
    for (auto i = 4; i <= 6; ++i) {
        dfs2(sum + i, c - 1, d, hist);
        dfs2(sum + i, c - 1, d, hist);
    }
}

int main() {
    int32_t n, k;
    cin >> n >> k;
    unordered_map<int32_t, int32_t> hist1, hist2;
    dfs(0, n, hist1);
    dfs2(0, k, n - k, hist2);
    int64_t sum = 0, win = 0;
    for (auto [k2, v2] : hist2) {
        for (auto [k1, v1] : hist1) {
            sum += int64_t(v1) * v2;
            if (k2 > k1)
                win += int64_t(v1) * v2;
        }
    }
    printf("%.6f\n", double(win) / sum);
    return 0;
}
0