結果

問題 No.75 回数の期待値の問題
ユーザー ciel
提出日時 2014-11-24 18:20:40
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,198 bytes
コンパイル時間 134 ms
コンパイル使用メモリ 12,288 KB
実行使用メモリ 18,432 KB
最終ジャッジ日時 2025-01-02 21:12:57
合計ジャッジ時間 18,400 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/python
#derived from checkio expected-dice

from fractions import Fraction
def gauss(a):
	if not a or len(a)==0: return None
	n=len(a)
	for i in range(n):
		if a[i][i]==0:
			for j in range(i+1,n):
				if a[j][i]!=0:
					for k in range(i,n+1): a[i][k]+=a[j][k]
					break
			else:
				return None
		for j in range(n):
			if i!=j:
				r = Fraction(a[j][i],a[i][i])
				for k in range(i,n+1): a[j][k] = a[j][k] - a[i][k]*r
	for i in range(n):
		x=Fraction(a[i][i],1)
		for j in range(len(a[i])):
			a[i][j] /= x
	return a

def expected(n, s, t): #, b):
	#perform probably-dice
	a=[0]*(s*(n+1)+1)
	for i in range(1,s+1):
		a[i+s]=Fraction(1,s)**n
	for e in range(n-1):
		for i in reversed(range(0,s*n+1)):
			a[i+s]=sum(a[i:i+s])
	l = t+1
	A = [[int(i==j) for i in range(l)]+[int(j!=t)] for j in range(l)]
	for start in range(l):
		if start != t:
			for roll in range(n,s*n+1):
				end = roll + start
				#end = (end + b[end%l])%l
				if end>=l: end=0
				A[start][end] -= a[roll+s]
	return float(gauss(A)[0][-1])

if __name__ == '__main__':
	import sys
	if sys.version_info[0]>=3: raw_input=input
	N=int(raw_input())
	print(expected(1,6,N))
0