結果
問題 | No.433 ICPC国内予選の選抜ルールがこんな感じだったらうれしい |
ユーザー | rpy3cpp |
提出日時 | 2016-10-21 13:13:38 |
言語 | Python3 (3.11.6 + numpy 1.26.0 + scipy 1.11.3) |
結果 |
AC
|
実行時間 | 484 ms / 4,000 ms |
コード長 | 1,259 bytes |
コンパイル時間 | 232 ms |
コンパイル使用メモリ | 10,996 KB |
実行使用メモリ | 38,800 KB |
最終ジャッジ日時 | 2023-08-15 10:24:43 |
合計ジャッジ時間 | 15,077 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 443 ms
38,536 KB |
testcase_01 | AC | 407 ms
38,272 KB |
testcase_02 | AC | 361 ms
36,880 KB |
testcase_03 | AC | 400 ms
35,236 KB |
testcase_04 | AC | 398 ms
34,316 KB |
testcase_05 | AC | 389 ms
37,360 KB |
testcase_06 | AC | 484 ms
38,800 KB |
testcase_07 | AC | 366 ms
37,736 KB |
testcase_08 | AC | 417 ms
35,740 KB |
testcase_09 | AC | 370 ms
26,192 KB |
testcase_10 | AC | 338 ms
24,396 KB |
testcase_11 | AC | 332 ms
24,436 KB |
testcase_12 | AC | 316 ms
23,712 KB |
testcase_13 | AC | 317 ms
24,084 KB |
testcase_14 | AC | 297 ms
23,656 KB |
testcase_15 | AC | 314 ms
23,832 KB |
testcase_16 | AC | 334 ms
24,336 KB |
testcase_17 | AC | 335 ms
24,428 KB |
testcase_18 | AC | 340 ms
24,504 KB |
testcase_19 | AC | 319 ms
24,244 KB |
testcase_20 | AC | 285 ms
23,448 KB |
testcase_21 | AC | 325 ms
24,148 KB |
testcase_22 | AC | 332 ms
24,516 KB |
testcase_23 | AC | 326 ms
24,444 KB |
testcase_24 | AC | 312 ms
24,216 KB |
testcase_25 | AC | 329 ms
24,144 KB |
testcase_26 | AC | 312 ms
24,064 KB |
testcase_27 | AC | 330 ms
24,188 KB |
testcase_28 | AC | 310 ms
24,160 KB |
testcase_29 | AC | 326 ms
24,368 KB |
testcase_30 | AC | 18 ms
9,396 KB |
testcase_31 | AC | 17 ms
9,360 KB |
testcase_32 | AC | 339 ms
24,688 KB |
testcase_33 | AC | 340 ms
24,384 KB |
testcase_34 | AC | 382 ms
28,644 KB |
testcase_35 | AC | 18 ms
9,392 KB |
testcase_36 | AC | 17 ms
9,352 KB |
testcase_37 | AC | 17 ms
9,284 KB |
testcase_38 | AC | 17 ms
9,492 KB |
testcase_39 | AC | 20 ms
9,440 KB |
testcase_40 | AC | 20 ms
9,400 KB |
testcase_41 | AC | 42 ms
10,816 KB |
testcase_42 | AC | 46 ms
10,676 KB |
testcase_43 | AC | 44 ms
10,872 KB |
testcase_44 | AC | 41 ms
10,848 KB |
testcase_45 | AC | 20 ms
9,400 KB |
testcase_46 | AC | 54 ms
11,008 KB |
testcase_47 | AC | 52 ms
11,008 KB |
ソースコード
import collections import heapq def read_data(): N, K = map(int, input().split()) db = [collections.defaultdict(list) for _ in range(11)] for teamID in range(N): s, p, u = map(int, input().split()) db[s][u].append((-p, teamID)) return N, K, db def solve(N, K, db): ans = [] counter = [0] * (10**5 + 1) for dbs in db[::-1]: if not dbs: continue result = solve_core(dbs, counter) ans.extend(result) if len(ans) >= K: return ans[:K] def solve_core(dbs, counter): ans = [] que = [] for univID, lst in dbs.items(): lst.sort() p = -lst[-1][0] teamID = lst[-1][1] que.append((counter[univID], p, teamID, univID)) del lst[-1] heapq.heapify(que) while que: counts, p, teamID, univID = que[0] ans.append(teamID) lst = dbs[univID] counter[univID] += 1 if lst: p = -lst[-1][0] teamID = lst[-1][1] heapq.heapreplace(que, (counts + 1, p, teamID, univID)) del dbs[univID][-1] else: heapq.heappop(que) return ans N, K, db = read_data() ans = solve(N, K, db) for univ_key in ans: print(univ_key)