結果
問題 | No.2453 Seat Allocation |
ユーザー | lloyz |
提出日時 | 2023-09-04 01:22:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 431 ms / 2,000 ms |
コード長 | 760 bytes |
コンパイル時間 | 183 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 110,544 KB |
最終ジャッジ日時 | 2024-06-25 09:37:46 |
合計ジャッジ時間 | 6,451 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
from heapq import heappop, heappush class T(object): def __init__(self, bunsi, bunbo, idx): self.bunsi = bunsi self.bunbo = bunbo self.idx = idx def __lt__(self, other): if other.bunbo * self.bunsi == self.bunbo * other.bunsi: return other.idx > self.idx else: return other.bunbo * self.bunsi > self.bunbo * other.bunsi n, m = map(int, input().split()) A = list(map(int, input().split())) B = list(map(int, input().split())) H = [] for i in range(n): heappush(H, T(A[i], B[0], i)) C = [0 for _ in range(n)] for _ in range(m): t = heappop(H) idx = t.idx print(idx + 1) C[idx] += 1 if C[idx] == m: continue heappush(H, T(A[idx], B[C[idx]], idx))