結果

問題 No.76 回数の期待値で練習
ユーザー lloyzlloyz
提出日時 2023-07-24 00:42:59
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 740 bytes
コンパイル時間 352 ms
コンパイル使用メモリ 81,968 KB
実行使用メモリ 846,868 KB
最終ジャッジ日時 2024-09-25 03:36:30
合計ジャッジ時間 3,413 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 55 ms
66,508 KB
testcase_01 MLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10**6)

E = [0, 1.0, 1.0833333333333333, 1.2569444444444444,
     1.5353009259259260, 1.6915991512345676, 2.0513639724794235]
P = [0.0 for _ in range(7)]
P[1] = E[2] - 1
P[2] = E[3] - 1 - P[1] * E[2]
P[3] = E[4] - 1 - P[1] * E[3] - P[2] * E[2]
P[4] = E[5] - 1 - P[1] * E[4] - P[2] * E[3] - P[3] * E[2]
P[5] = E[6] - 1 - P[1] * E[5] - P[2] * E[4] - P[3] * E[3] - P[4] * E[2]
P[6] = 1.0 - sum(P)

def rec(x):
    if x in memo:
        return memo[x]
    res = 1.0
    for i in range(1, 7):
        y = x - i
        if y < 0:
            y = 0
        res += rec(y) * P[i]
    memo[x] = res
    return res


for _ in range(int(input())):
    n = int(input())
    memo = {}
    memo[0] = 0.0
    print(rec(n))
0