結果

問題 No.58 イカサマなサイコロ
ユーザー 🍡yurahuna🍡yurahuna
提出日時 2016-02-25 20:02:39
言語 Python3
(3.11.6 + numpy 1.26.0 + scipy 1.11.3)
結果
AC  
実行時間 31 ms / 5,000 ms
コード長 1,215 bytes
コンパイル時間 840 ms
コンパイル使用メモリ 11,868 KB
実行使用メモリ 10,200 KB
最終ジャッジ日時 2023-10-23 20:05:50
合計ジャッジ時間 1,853 ms
ジャッジサーバーID
(参考情報)
judge11 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 27 ms
10,180 KB
testcase_01 AC 31 ms
10,200 KB
testcase_02 AC 27 ms
10,180 KB
testcase_03 AC 27 ms
10,180 KB
testcase_04 AC 28 ms
10,184 KB
testcase_05 AC 29 ms
10,184 KB
testcase_06 AC 29 ms
10,192 KB
testcase_07 AC 28 ms
10,180 KB
testcase_08 AC 28 ms
10,180 KB
testcase_09 AC 30 ms
10,192 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

# -*- coding: utf-8 -*-

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

max_sum = 6 * N

# Taro
# dp1[num_used][cur_sum] = num of such pattern
# 普通のサイコロ→イカサマサイコロの順で考える
dp1 = [[0] * (max_sum + 1) for i in range(N + 1)]

if N - K > 0:
    # 普通のサイコロがあるとき
    for j in range(1, 6 + 1):
        dp1[1][j] = 1
else:
    for j in range(4, 6 + 1):
        dp1[1][j] = 2

# 配るDP
for n in range(1, N - K):
    for s in range(1, max_sum + 1):
        for d in range(1, 6 + 1):
            if s + d <= max_sum:
                dp1[n + 1][s + d] += dp1[n][s]

for n in range(N - K, N):
    for s in range(1, max_sum + 1):
        for d in range(4, 6 + 1):
            if s + d <= max_sum:
                dp1[n + 1][s + d] += 2 * dp1[n][s]

# Jiro
dp2 = [[0] * (max_sum + 1) for i in range(N + 1)]
for j in range(1, 6 + 1):
    dp2[1][j] = 1

for n in range(1, N):
    for s in range(1, max_sum + 1):
        for d in range(1, 6 + 1):
            if s + d <= max_sum:
                dp2[n + 1][s + d] += dp2[n][s]

cumu2 = 0
ans = 0
for s in range(2, max_sum + 1):
    cumu2 += dp2[N][s-1]
    ans += dp1[N][s] * cumu2

for i in range(2 * N):
    ans /= 6

print(ans)
0