結果

問題 No.75 回数の期待値の問題
ユーザー cielciel
提出日時 2014-11-24 18:20:40
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,198 bytes
コンパイル時間 362 ms
コンパイル使用メモリ 10,960 KB
実行使用メモリ 15,544 KB
最終ジャッジ日時 2023-08-30 22:43:48
合計ジャッジ時間 8,923 ms
ジャッジサーバーID
(参考情報)
judge14 / judge11
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 34 ms
9,948 KB
testcase_01 AC 30 ms
9,792 KB
testcase_02 AC 30 ms
9,996 KB
testcase_03 AC 32 ms
9,948 KB
testcase_04 AC 29 ms
9,872 KB
testcase_05 AC 30 ms
9,976 KB
testcase_06 AC 30 ms
9,872 KB
testcase_07 AC 32 ms
10,036 KB
testcase_08 AC 33 ms
9,960 KB
testcase_09 AC 33 ms
10,024 KB
testcase_10 AC 34 ms
9,868 KB
testcase_11 AC 36 ms
9,856 KB
testcase_12 AC 37 ms
9,976 KB
testcase_13 AC 37 ms
9,864 KB
testcase_14 AC 40 ms
9,960 KB
testcase_15 AC 52 ms
9,860 KB
testcase_16 AC 1,250 ms
10,280 KB
testcase_17 TLE -
testcase_18 -- -
testcase_19 -- -
権限があれば一括ダウンロードができます

ソースコード

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