結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
コンテスト
ユーザー tktk_snsn
提出日時 2021-01-14 01:34:45
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 332 ms / 4,000 ms
コード長 819 bytes
コンパイル時間 154 ms
コンパイル使用メモリ 81,992 KB
実行使用メモリ 93,092 KB
最終ジャッジ日時 2024-11-22 23:37:26
合計ジャッジ時間 11,425 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from collections import defaultdict
import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)
MAX = 10**5


def main():
    N, K = map(int, input().split())

    solve = [[] for _ in range(11)]
    for i in range(N):
        S, P, U = map(int, input().split())
        solve[S].append((U, P, i))

    selected = [0] * (MAX + 1)
    for team in solve[::-1]:
        if not team:
            continue
        memo = defaultdict(list)
        team.sort()
        for u, p, i in team:
            memo[selected[u]].append((p, i))
            selected[u] += 1

        keys = sorted(memo.keys())
        for key in keys:
            val = sorted(memo[key])
            for _, i in val:
                print(i)
                K -= 1
                if not K:
                    return
    return


main()
0