結果

問題 No.58 イカサマなサイコロ
コンテスト
ユーザー norioc
提出日時 2025-02-05 02:51:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 68 ms / 5,000 ms
コード長 732 bytes
コンパイル時間 765 ms
コンパイル使用メモリ 82,728 KB
実行使用メモリ 60,032 KB
最終ジャッジ日時 2025-02-05 02:51:38
合計ジャッジ時間 2,476 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
K = int(input())


def roll(dice_getter):
    dp = [0] * (6 * N + 1)
    dp[0] = 1
    for i in range(N):
        dice = dice_getter(i)
        pp = [0] * (6 * N + 1)
        dp, pp = pp, dp
        for j in range(6 * N + 1):
            if pp[j] == 0: continue
            for d in dice:
                dp[j+d] += pp[j]

    return dp


fair_dice = [1, 2, 3, 4, 5, 6]
loaded_dice = [4, 4, 5, 5, 6, 6]
dice_sum1 = roll(lambda p: fair_dice)
dice_sum2 = roll(lambda p: fair_dice if p >= K else loaded_dice)

win = 0
tot = 0
for i in range(1, 6*N+1):
    for j in range(1, 6*N+1):
        tot += dice_sum1[i] * dice_sum2[j]
        if i < j:
            win += dice_sum1[i] * dice_sum2[j]

ans = win / tot
print(ans)
0