結果

問題 No.301 サイコロで確率問題 (1)
ユーザー rlangevinrlangevin
提出日時 2023-10-29 14:49:36
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 752 ms / 1,000 ms
コード長 612 bytes
コンパイル時間 109 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-09-25 16:42:46
合計ジャッジ時間 2,840 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

diff #

def solve(K):
    def check(m):
        dp = [1.0] * (K + 2)
        dp[-1] = m
        dp[K] = 0.0
        for i in range(K - 1, -1, -1):
            for j in range(1, 7):
                dp[i] += dp[min(i + j, K + 1)]/6
                
        return dp[0] <= dp[-1]

    yes = 1e3
    no = 0
    for _ in range(60):
        mid = (yes + no)/2
        if check(mid):
            yes = mid
        else:
            no = mid
            
    return yes
    
T = int(input())
dp = [0] * 100
for i in range(100):
    dp[i] = solve(i) - i
    
for _ in range(T):
    N = int(input())
    print(N + dp[min(99, N)])
0