結果

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

ソースコード

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