結果

問題 No.759 悪くない忘年会にしような!
ユーザー tktk_snsn
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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")
0