結果

問題 No.58 イカサマなサイコロ
ユーザー cherrypi59cherrypi59
提出日時 2018-10-07 21:58:27
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 31 ms / 5,000 ms
コード長 866 bytes
コンパイル時間 95 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-10-12 14:18:55
合計ジャッジ時間 1,056 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

diff #

def main():
    n, k = int(input()), int(input())

    dp = [[0 for j in range(6*n+1)] for i in range(n+1)]
    dp[0][0] = 1

    for i in range(1, n+1):
        for j in range(i, 6*i+1):
            for l in range(1, 7):
                if l <= j:
                    dp[i][j] += dp[i-1][j-l]

    dp2 = [[0 for j in range(6*n+1)] for i in range(k+1)]
    dp2[0][0] = 1

    for i in range(1, k+1):
        for j in range(4*i, 6*i+1):
            for l in range(4, 7):
                if l <= j:
                    dp2[i][j] += dp2[i-1][j-l]

    dp3 = [0] * (6*n+1)
    for i in range(6*n+1):
        for j in range(i+1):
            dp3[i] += dp[n-k][j] * dp2[k][i-j]

    comp, win = sum(dp[n]) * sum(dp3), 0
    for i in range(6*n+1):
        for j in range(i):
            win += dp3[i] * dp[n][j]

    print(win/comp)


if __name__ == '__main__':
    main()
0