結果

問題 No.1285 ゴミ捨て
ユーザー kohei2019kohei2019
提出日時 2020-12-13 00:03:01
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 523 ms / 2,000 ms
コード長 4,690 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 13,184 KB
実行使用メモリ 15,460 KB
最終ジャッジ日時 2024-05-06 14:43:04
合計ジャッジ時間 6,175 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 28 ms
11,008 KB
testcase_01 AC 27 ms
11,008 KB
testcase_02 AC 26 ms
11,136 KB
testcase_03 AC 27 ms
11,136 KB
testcase_04 AC 28 ms
11,008 KB
testcase_05 AC 28 ms
11,008 KB
testcase_06 AC 26 ms
11,136 KB
testcase_07 AC 513 ms
15,360 KB
testcase_08 AC 42 ms
11,264 KB
testcase_09 AC 223 ms
12,928 KB
testcase_10 AC 440 ms
14,720 KB
testcase_11 AC 398 ms
14,464 KB
testcase_12 AC 293 ms
13,696 KB
testcase_13 AC 303 ms
13,696 KB
testcase_14 AC 477 ms
15,100 KB
testcase_15 AC 321 ms
13,952 KB
testcase_16 AC 105 ms
11,904 KB
testcase_17 AC 105 ms
11,776 KB
testcase_18 AC 99 ms
11,648 KB
testcase_19 AC 201 ms
12,800 KB
testcase_20 AC 93 ms
11,648 KB
testcase_21 AC 141 ms
12,032 KB
testcase_22 AC 523 ms
15,456 KB
testcase_23 AC 520 ms
15,460 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class BalancingTree:
    def __init__(self, n):
        self.N = n
        self.root = self.node(1<<n, 1<<n)

    def debug(self):
        def debug_info(nd_):
            return (nd_.value - 1, nd_.pivot - 1, nd_.left.value - 1 if nd_.left else -1, nd_.right.value - 1 if nd_.right else -1)

        def debug_node(nd):
            re = []
            if nd.left:
                re += debug_node(nd.left)
            if nd.value: re.append(debug_info(nd))
            if nd.right:
                re += debug_node(nd.right)
            return re
        print("Debug - root =", self.root.value - 1, debug_node(self.root)[:50])

    def append(self, v):# v を追加（その時点で v はない前提）
        v += 1
        nd = self.root
        while True:
            if v == nd.value:
                # v がすでに存在する場合に何か処理が必要ならここに書く
                return 0
            else:
                mi, ma = min(v, nd.value), max(v, nd.value)
                if mi < nd.pivot:
                    nd.value = ma
                    if nd.left:
                        nd = nd.left
                        v = mi
                    else:
                        p = nd.pivot
                        nd.left = self.node(mi, p - (p&-p)//2)
                        break
                else:
                    nd.value = mi
                    if nd.right:
                        nd = nd.right
                        v = ma
                    else:
                        p = nd.pivot
                        nd.right = self.node(ma, p + (p&-p)//2)
                        break

    def leftmost(self, nd):
        if nd.left: return self.leftmost(nd.left)
        return nd

    def rightmost(self, nd):
        if nd.right: return self.rightmost(nd.right)
        return nd

    def find_l(self, v): # vより真に小さいやつの中での最大値（なければ-1）
        v += 1
        nd = self.root
        prev = 0
        if nd.value < v: prev = nd.value
        while True:
            if v <= nd.value:
                if nd.left:
                    nd = nd.left
                else:
                    return prev - 1
            else:
                prev = nd.value
                if nd.right:
                    nd = nd.right
                else:
                    return prev - 1

    def find_r(self, v): # vより真に大きいやつの中での最小値（なければRoot）
        v += 1
        nd = self.root
        prev = 0
        if nd.value > v: prev = nd.value
        while True:
            if v < nd.value:
                prev = nd.value
                if nd.left:
                    nd = nd.left
                else:
                    return prev - 1
            else:
                if nd.right:
                    nd = nd.right
                else:
                    return prev - 1

    @property
    def max(self):
        return self.find_l((1<<self.N)-1)

    @property
    def min(self):
        return self.find_r(-1)

    def delete(self, v, nd = None, prev = None): # 値がvのノードがあれば削除（なければ何もしない）
        v += 1
        if not nd: nd = self.root
        if not prev: prev = nd
        while v != nd.value:
            prev = nd
            if v <= nd.value:
                if nd.left:
                    nd = nd.left
                else:
                    return
            else:
                if nd.right:
                    nd = nd.right
                else:
                    return
        if (not nd.left) and (not nd.right):
            if nd.value < prev.value:
                prev.left = None
            else:
                prev.right = None
        elif not nd.left:
            if nd.value < prev.value:
                prev.left = nd.right
            else:
                prev.right = nd.right
        elif not nd.right:
            if nd.value < prev.value:
                prev.left = nd.left
            else:
                prev.right = nd.left
        else:
            nd.value = self.leftmost(nd.right).value
            self.delete(nd.value - 1, nd.right, nd)

    def __contains__(self, v: int) -> bool:
        return self.find_r(v - 1) == v

    class node:
        def __init__(self, v, p):
            self.value = v
            self.pivot = p
            self.left = None
            self.right = None

N = int(input())
lsA = [int(input()) for i in range(N)]
lsA.sort()
lsw = BalancingTree(35)
ans = 0
for i in range(N):
    a = lsw.find_l(lsA[i]-1)
    if a == -1:
        ans += 1
        lsw.append(lsA[i])
    else:
        lsw.delete(a)
        lsw.append(lsA[i])
print(ans)
0