結果

問題 No.76 回数の期待値で練習
ユーザー lloyzlloyz
提出日時 2023-07-24 00:42:44
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 740 bytes
コンパイル時間 202 ms
コンパイル使用メモリ 12,032 KB
実行使用メモリ 240,576 KB
最終ジャッジ日時 2023-10-25 08:28:11
合計ジャッジ時間 6,892 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

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