結果

問題 No.75 回数の期待値の問題
コンテスト
ユーザー sjiki
提出日時 2024-08-06 06:00:05
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 553 bytes
コンパイル時間 534 ms
コンパイル使用メモリ 82,660 KB
実行使用メモリ 67,404 KB
最終ジャッジ日時 2024-08-06 06:00:09
合計ジャッジ時間 2,490 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 4
other RE * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    import sys
    input = sys.stdin.read
    from scipy.linalg import solve
    
    K = int(input().strip())
    
    # 期待値の方程式を構築
    A = [[0] * K for _ in range(K)]
    B = [0] * K
    
    for i in range(K):
        A[i][i] = 1
        for j in range(1, 7):
            if i + j < K:
                A[i][i + j] -= 1 / 6
            else:
                B[i] += 1 / 6
    
    # 連立方程式を解く
    E = solve(A, B)
    
    # 結果を出力
    print(f"{E[0]:.10f}")

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