結果

問題 No.649 ここでちょっとQK！
ユーザー toyuzukotoyuzuko
提出日時 2020-06-06 12:34:07
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,643 ms / 3,000 ms
コード長 4,606 bytes
コンパイル時間 314 ms
コンパイル使用メモリ 87,284 KB
実行使用メモリ 102,472 KB
最終ジャッジ日時 2023-08-25 00:29:22
合計ジャッジ時間 24,429 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 71 ms
71,488 KB
testcase_01 AC 72 ms
71,444 KB
testcase_02 AC 71 ms
71,332 KB
testcase_03 AC 141 ms
87,592 KB
testcase_04 AC 238 ms
95,348 KB
testcase_05 AC 312 ms
97,104 KB
testcase_06 AC 241 ms
94,452 KB
testcase_07 AC 72 ms
71,432 KB
testcase_08 AC 70 ms
71,264 KB
testcase_09 AC 71 ms
71,328 KB
testcase_10 AC 71 ms
71,516 KB
testcase_11 AC 71 ms
71,540 KB
testcase_12 AC 870 ms
91,352 KB
testcase_13 AC 872 ms
91,516 KB
testcase_14 AC 850 ms
90,132 KB
testcase_15 AC 969 ms
92,324 KB
testcase_16 AC 923 ms
91,216 KB
testcase_17 AC 971 ms
90,936 KB
testcase_18 AC 1,110 ms
94,792 KB
testcase_19 AC 1,161 ms
97,592 KB
testcase_20 AC 1,166 ms
95,020 KB
testcase_21 AC 1,287 ms
98,632 KB
testcase_22 AC 1,330 ms
98,748 KB
testcase_23 AC 1,444 ms
100,196 KB
testcase_24 AC 1,467 ms
100,612 KB
testcase_25 AC 1,548 ms
100,864 KB
testcase_26 AC 1,643 ms
102,472 KB
testcase_27 AC 188 ms
78,176 KB
testcase_28 AC 197 ms
78,320 KB
testcase_29 AC 193 ms
78,392 KB
testcase_30 AC 918 ms
92,012 KB
testcase_31 AC 1,027 ms
92,724 KB
testcase_32 AC 73 ms
71,300 KB
testcase_33 AC 71 ms
71,420 KB
testcase_34 AC 71 ms
71,416 KB
testcase_35 AC 70 ms
71,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class SplayNode():
    def __init__(self, val):
        self.val = val
        self.par = None
        self.lt = None
        self.rt = None
        self.size = 1

class SplayTree():
    def __init__(self):
        self.size = 0
        self.root = None

    def update(self, node):
        node.size = 1
        if node.lt is not None:
            node.size += node.lt.size
        if node.rt is not None:
            node.size += node.rt.size

    def rotate(self, node):
        par = node.par
        if par is None:
            return
        ppar = par.par
        if par.lt == node:
            rt = node.rt
            par.lt = rt
            par.par = node
            node.rt = par
            self.update(par)
            self.update(node)
            if rt is not None:
                rt.par = par
            node.par = ppar
            if ppar is None:
                return
            if ppar.lt == par:
                ppar.lt = node
            else:
                ppar.rt = node
        else:
            lt = node.lt
            par.rt = lt
            par.par = node
            node.lt = par
            self.update(par)
            self.update(node)
            if lt is not None:
                lt.par = par
            node.par = ppar
            if ppar is None:
                return
            if ppar.rt == par:
                ppar.rt = node
            else:
                ppar.lt = node

    def splay(self, node):
        while node.par is not None:
            self.rotate(node)
        self.size = node.size
        self.root = node

    def get(self, k):
        node = self.root
        if self.size <= k:
            return False
        while True:
            lsize = 0 if node.lt is None else node.lt.size
            if lsize == k:
                self.splay(node)
                return node
            if lsize > k:
                node = node.lt
            else:
                k -= lsize + 1
                node = node.rt

    def find(self, val):
        node = self.root
        if node is None:
            return False, 0
        count = 0
        while True:
            lsize = 0 if node.lt is None else node.lt.size
            if node.val == val:
                self.splay(node)
                return True, count + lsize
            if node.val > val:
                if node.lt is None:
                    return False, count + lsize
                else:
                    node = node.lt
            else:
                if node.rt is None:
                    return False, count + lsize + 1
                else:
                    count += lsize + 1
                    node = node.rt

    def merge(self, other):
        if self.size == 0:
            self.size = other.size
            self.root = other.root
            return
        if other.size == 0:
            return
        lnode = self.get(self.size - 1)
        rnode = other.get(0)
        rnode.par = lnode
        lnode.rt = rnode
        self.update(lnode)
        self.root = lnode
        self.size = lnode.size

    def split(self, k):
        other = SplayTree()
        if k == 0:
            other.root = self.root
            other.size = self.size
            self.size = 0
            self.root = None
            return other
        if k == self.size:
            return other
        root = self.get(k)
        lroot = root.lt
        rroot = root
        rroot.lt = None
        lroot.par = None
        self.update(lroot)
        self.update(rroot)
        self.size = lroot.size
        self.root = lroot
        other.size = rroot.size
        other.root = rroot
        return other

    def insert(self, val):
        _, k = self.find(val)
        rtree = self.split(k)
        nnode = SplayNode(val)
        ntree = SplayTree()
        ntree.root = nnode
        ntree.size = 1
        self.merge(ntree)
        self.merge(rtree)

    def delete(self, val):
        exist, k = self.find(val)
        if not exist:
            return
        rtree = self.split(k + 1)
        deleted = self.split(k)
        self.merge(rtree)

    def delete_kth(self, k):
        rtree = self.split(k + 1)
        deleted = self.split(k)
        self.merge(rtree)


import sys
input = sys.stdin.readline

Q, K = map(int, input().split())

st = SplayTree()
res = []

for _ in range(Q):
    q = tuple(map(int, input().split()))
    if q[0] == 1:
        st.insert(q[1])
    else:
        v = st.get(K - 1)
        if not v:
            res.append(-1)
        else:
            res.append(v.val)
            st.delete_kth(K - 1)

print('\n'.join(map(str, res)))
0