結果

問題 No.58 イカサマなサイコロ
ユーザー 🍡yurahuna
提出日時 2016-02-25 20:02:39
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

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