結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
コンテスト
ユーザー roaris
提出日時 2019-12-14 20:58:57
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 399 ms / 4,000 ms
+ 478µs
コード長 580 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,915 ms
コンパイル使用メモリ 96,108 KB
実行使用メモリ 115,068 KB
最終ジャッジ日時 2026-07-31 15:22:19
合計ジャッジ時間 20,622 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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