結果
問題 |
No.759 悪くない忘年会にしような!
|
ユーザー |
![]() |
提出日時 | 2022-05-12 12:27:11 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,472 bytes |
コンパイル時間 | 769 ms |
コンパイル使用メモリ | 82,108 KB |
実行使用メモリ | 157,312 KB |
最終ジャッジ日時 | 2024-07-20 05:31:17 |
合計ジャッジ時間 | 15,543 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 WA * 36 |
ソースコード
from collections import deque import sys input = sys.stdin.buffer.readline sys.setrecursionlimit(10 ** 7) N = int(input()) PQR = tuple(tuple(map(int, input().split())) for _ in range(N)) P, Q, R = zip(*PQR) ptoi = {p: i for i, p in enumerate(sorted(set(P)))} M = len(ptoi) d = [[] for _ in range(M)] for i, (p, q, r) in enumerate(PQR, 1): d[ptoi[p]].append((q, r, i)) for i in range(M): stack = [] d[i].sort() for q, r, j in d[i]: if stack and stack[-1][1] <= r: stack.pop() stack.append((q, r, j)) d[i] = stack d[i].reverse() def merge(i, j): F = d[i] G = d[j] stack = [] pri = [] def insert(x): while stack: if pri[-1] == 0 and stack[-1][1] <= x[1]: stack.pop() pri.pop() continue break stack.append(x) pri.append(1) while F and G: if F[-1][0] <= G[-1][0]: stack.append(F.pop()) pri.append(0) else: insert(G.pop()) while F: stack.append(F.pop()) pri.append(0) while G: insert(G.pop()) stack.reverse() d[i] = stack que = deque(range(M)) while len(que) > 1: a = que.popleft() b = que.popleft() if a > b: que.append(a) que.appendleft(b) continue merge(a, b) que.append(a) ans = [] for _, _, i in d[0]: ans.append(i) ans.sort() print(*ans, sep="\n")