結果

問題 No.463 魔法使いのすごろく🎲
コンテスト
ユーザー compass19
提出日時 2016-12-14 00:48:40
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 527 bytes
コンパイル時間 171 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-11-30 01:58:04
合計ジャッジ時間 2,911 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 10 RE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

n, m = map(int, input().split())
c = list(map(int, input().split()))

def move_prob(pos, P_list):
	if pos == n-1:   # 最後の1回はズル
		return sum(P_list[-1-m:])
	prob = sum(P_list[max(0, pos-m):min(n-m-1, pos)]) / m
	return prob

OLD_P = [1] + [0]*(n-1)
EXP_GOLD = 0
for _ in range(n-m):
	# サイコロを振って確率の更新
	NEW_P = [move_prob(pos, OLD_P) for pos in range(n)]
	# 部分期待値の計算
	subgold = sum([i*j for i, j in zip(c, NEW_P[1:-1])])
	EXP_GOLD += subgold
	OLD_P = NEW_P

print(EXP_GOLD)
0