結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
ユーザー roaris
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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()
0