結果

問題 No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい
コンテスト
ユーザー maspy
提出日時 2020-03-18 22:16:09
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 644 ms / 4,000 ms
コード長 679 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 207 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 56,932 KB
最終ジャッジ日時 2024-12-14 02:35:21
合計ジャッジ時間 25,839 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#!/usr/bin/env python3.8
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
from operator import itemgetter

N, K = map(int, readline().split())
m = map(int, read().split())
S, P, U = zip(*zip(m, m, m))

SP = [-(s << 20) + p for s, p in zip(S, P)]
rank_1, _ = zip(*sorted(enumerate(SP), key=itemgetter(1)))
univ_cnt = [0] * (10 ** 6 + 10)
rank_in_univ = [0] * N
for i in rank_1:
    rank_in_univ[i] = univ_cnt[U[i]]
    univ_cnt[U[i]] += 1

key = (-(s << 40) + (r << 20) + p for s, r, p in zip(S, rank_in_univ, P))
answer, _ = zip(*sorted(enumerate(key), key=itemgetter(1)))
print('\n'.join(map(str, answer[:K])))
0