結果

問題 No.75 回数の期待値の問題
コンテスト
ユーザー nohtaray
提出日時 2019-11-18 13:49:00
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 601 ms / 5,000 ms
コード長 457 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 44,188 KB
最終ジャッジ日時 2024-10-01 13:52:16
合計ジャッジ時間 14,036 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import os
import sys

import numpy as np

if os.getenv("LOCAL"):
    sys.stdin = open("_in.txt", "r")

sys.setrecursionlimit(10 ** 9)
INF = float("inf")
IINF = 10 ** 18
MOD = 10 ** 9 + 7
# MOD = 998244353


K = int(sys.stdin.buffer.readline())

# X[i] = sum(X[i+1:i+7]) + 1
# 反復法で解く
ans = np.ones(K + 10)
ans[K] = 0
for _ in range(10000):
    cs = ans.cumsum()
    ans[:K] = (cs[6:] - cs[:-6])[:K] / 6 + 1
    ans[K + 1:] = ans[0]
print(ans[0])
0