結果

問題 No.58 イカサマなサイコロ
ユーザー 🍡yurahuna🍡yurahuna
提出日時 2016-02-25 20:02:39
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 32 ms / 5,000 ms
コード長 1,215 bytes
コンパイル時間 248 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-09-22 13:45:56
合計ジャッジ時間 1,223 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 29 ms
10,880 KB
testcase_01 AC 31 ms
10,880 KB
testcase_02 AC 30 ms
11,008 KB
testcase_03 AC 30 ms
10,880 KB
testcase_04 AC 30 ms
10,880 KB
testcase_05 AC 31 ms
11,008 KB
testcase_06 AC 32 ms
10,880 KB
testcase_07 AC 30 ms
10,880 KB
testcase_08 AC 29 ms
10,880 KB
testcase_09 AC 30 ms
10,880 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