結果

問題 No.2220 Range Insert & Point Mex
ユーザー rlangevinrlangevin
提出日時 2023-02-17 22:49:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 679 ms / 2,000 ms
コード長 1,496 bytes
コンパイル時間 487 ms
コンパイル使用メモリ 82,500 KB
実行使用メモリ 136,320 KB
最終ジャッジ日時 2024-07-21 12:51:29
合計ジャッジ時間 14,486 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
readline = sys.stdin.readline
from heapq import *
from collections import *

class QuasiBinaryTree:
    def __init__(self, rule="min"):
        if rule == "max":
            self.pm = -1
        else:
            self.pm = 1
 
        self.inf = 10**18
        self.P = [self.inf]
        self.Q = []
 
    def insert(self, x):
        heappush(self.P, x * self.pm)
 
    def erase(self, x):
        heappush(self.Q, x * self.pm)
 
    def top(self):
        while self.Q and (self.P[0] == self.Q[0]):
            heappop(self.P)
            heappop(self.Q)
        return self.P[0] * self.pm

def f(r, a):
    return r * M + a

N = int(readline())
M = 10 ** 9 + 7
lst = []
for i in range(N):
    L, R, A = map(int, readline().split())
    lst.append((L, R, A))
    
lst.sort(reverse=True)
Q = int(readline())
X = list(map(int, readline().split()))

H = []
S = QuasiBinaryTree("min")
for i in range(101010):
    S.insert(i)
    
cnt = defaultdict(int)
for x in X:
    while lst:
        L, R, A = lst[-1]
        if A >= 100010:
            lst.pop()
            continue
        if L <= x:
            heappush(H, (R, A))
            cnt[A] += 1
            if cnt[A] == 1:
                S.erase(A)
            lst.pop()
        else:
            break
            
    while H:
        R, A = H[0]
        if R < x:
            cnt[A] -= 1
            if cnt[A] == 0:
                S.insert(A)
            heappop(H)
        else:
            break
           
    print(S.top())
0