結果

問題 No.2390 Udon Coupon (Hard)
ユーザー 👑 Mizar
提出日時 2023-07-27 02:05:12
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 546 ms / 2,000 ms
コード長 854 bytes
コンパイル時間 126 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 44,736 KB
最終ジャッジ日時 2024-10-06 02:06:12
合計ジャッジ時間 25,307 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 47
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
n = int(input())
[[a1, b1], [a2, b2], [a3, b3]] = [list(map(int, input().split())) for _ in range(3)]
# 最大効率の割引を a1,b1 とする
if a2 * b1 < a1 * b2:
	a1, b1, a2, b2 = a2, b2, a1, b1
if a3 * b1 < a1 * b3:
	a1, b1, a3, b3 = a3, b3, a1, b1
# a1inv : ceil(2^34 / a1)
a1inv = ((2**34 - 1) // a1) + 1
# c1min : a1,b1 の割引の最低使用回数
c1min = max(n // a1 - a2 - a3, 0)
# remain : c1min回を除いた残り枚数
remain = min(a1 * (a2 + a3) + (n % a1), n)
print(
	max(
		# t * a1 < 2^34  -->  floor(ceil(2^34 / a1) * t / 2^34)
		# (t * a1inv >> 34) == (t // a1)
		int(np.max(((s - j * a3) * a1inv >> 34) * b1 + i * b2 + j * b3))
		for i, j, s in zip(
			range(a1),
			map(lambda s: np.arange(0, min(a1, s // a3 + 1), dtype=np.uint64), range(remain, -1, -a2)),
			range(remain, -1, -a2),
		)
	)
	+ c1min * b1
)
0