結果

問題 No.2389 Cheating Code Golf
ユーザー shobonvipshobonvip
提出日時 2023-07-21 23:19:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,647 ms / 2,000 ms
コード長 1,437 bytes
コンパイル時間 197 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 77,952 KB
最終ジャッジ日時 2024-09-22 00:51:36
合計ジャッジ時間 82,241 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

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

if n == 0:
	print(ans)
	exit()

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