結果

問題 No.58 イカサマなサイコロ
ユーザー ooooobaoooooba
提出日時 2021-09-23 21:00:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,974 ms / 5,000 ms
コード長 1,249 bytes
コンパイル時間 1,834 ms
コンパイル使用メモリ 121,288 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-09-18 19:57:39
合計ジャッジ時間 4,716 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 1,974 ms
4,380 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 2 ms
4,380 KB
testcase_04 AC 3 ms
4,380 KB
testcase_05 AC 11 ms
4,376 KB
testcase_06 AC 334 ms
4,380 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 2 ms
4,376 KB
testcase_09 AC 326 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

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