結果

問題 No.2389 Cheating Code Golf
コンテスト
ユーザー shobonvip
提出日時 2023-07-21 23:19:15
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,405 bytes
コンパイル時間 144 ms
コンパイル使用メモリ 82,240 KB
実行使用メモリ 78,172 KB
最終ジャッジ日時 2024-09-22 00:49:47
合計ジャッジ時間 82,225 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 45 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

import random
import time

nn, m = map(int,input().split())
d = []

st = time.time()

n = 0
ans = 0
nans = 0
for i in range(nn):
	a, b, p = map(int,input().split())
	if a <= b:
		ans += 1 / a
		nans += 1 / a
	elif p == 1:
		ans += 1 / b
		nans += 1 / b
	else:
		d.append((a, b, p))
		n += 1

d.sort(key= lambda x:(-(x[0]-x[1])/(x[1]*x[0]*x[2]), x[0]))

dp = [0.0] * (n+1)
dp[0] = 1.0
for i in range(m):
	ndp = [0.0] * (n+1)
	for j in range(n+1):
		tar = 1.0
		for k in range(j, n):
			ndp[k] += dp[j] * tar * (1 - 1/d[k][2])
			tar *= 1/d[k][2]
		ndp[n] += dp[j] * tar
	dp = ndp

tmp = 0.0
for i in range(n): tmp += 1 / d[i][0]

#print(tmp)
for i in range(n+1):
	ans += tmp * dp[i]
	if i < n:
		tmp -= 1 / d[i][0]
		tmp += 1 / d[i][1]
	#print(tmp, dp[i])

nnans = nans
dd = d[::]

while time.time() - st < 1.6:
	d = dd[::]
	for _ in range(2):
		a, b = random.randrange(n), random.randrange(n)
		r = d[a]
		d[a] = d[b]
		d[b] = r
		
	nans = nnans
	dp = [0.0] * (n+1)
	dp[0] = 1.0
	for i in range(m):
		ndp = [0.0] * (n+1)
		for j in range(n+1):
			tar = 1.0
			for k in range(j, n):
				ndp[k] += dp[j] * tar * (1 - 1/d[k][2])
				tar *= 1/d[k][2]
			ndp[n] += dp[j] * tar
		dp = ndp

	tmp = 0.0
	for i in range(n): tmp += 1 / d[i][0]

	#print(tmp)
	for i in range(n+1):
		nans += tmp * dp[i]
		if i < n:
			tmp -= 1 / d[i][0]
			tmp += 1 / d[i][1]
		#print(tmp, dp[i])

	ans = max(ans, nans)

	
print(ans)
0