結果

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

ソースコード

diff # 

n, m = map(int, input().split())
if n == 2:
	print(0); exit()
c = list(map(int, input().split()))

def move_prob(pos, P_list):
	if pos == n-1:   # 最後の1回はズル
		return sum(P_list[-1-m:])
	# print(pos, P_list[max(0, pos-m):min(n-m-1, pos)])
	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+1):
	# サイコロを振って確率の更新
	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('subgold', subgold)
	print(OLD_P)

print(EXP_GOLD)
0