結果
問題 |
No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
|
ユーザー |
|
提出日時 | 2019-12-14 20:58:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 645 ms / 4,000 ms |
コード長 | 580 bytes |
コンパイル時間 | 209 ms |
コンパイル使用メモリ | 82,156 KB |
実行使用メモリ | 106,112 KB |
最終ジャッジ日時 | 2024-06-28 03:24:26 |
合計ジャッジ時間 | 24,074 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 48 |
ソースコード
from heapq import * N, K = map(int, input().split()) SPU = [tuple(map(int, input().split()))+(i,) for i in range(N)] SPU.sort(key=lambda k: (-k[0], k[1])) SPU += [(-1, -1, -1, -1)] pq = [] cnt = [0]*N ans = [] for i in range(N): Si, Pi, Ui, idx = SPU[i] heappush(pq, (cnt[Ui], Pi, idx)) cnt[Ui] += 1 if SPU[i][0]!=SPU[i+1][0]: while pq: _, _, idx = heappop(pq) ans.append(idx) if len(ans)>=K: for ans_i in ans: print(ans_i) exit()