結果

問題 No.3134 二分探索木
コンテスト
ユーザー 👑 cologne
提出日時 2026-09-29 15:19:34
言語 Python3
(3.14.7 + numpy 2.5.2 + scipy 1.18.0 + ACL)
コンパイル:
python3 -mpy_compile _filename_
実行:
python3 _filename_
結果
AC  
実行時間 449 ms / 2,000 ms
+ 890µs
コード長 1,417 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 55 ms
コンパイル使用メモリ 14,976 KB
実行使用メモリ 116,564 KB
最終ジャッジ日時 2026-09-29 15:19:42
合計ジャッジ時間 6,973 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge4_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
from typing import List, Tuple
sys.setrecursionlimit(10**6)

MOD = 998244353

int1 = lambda x: int(x) - 1
input = lambda: sys.stdin.readline().rstrip('\n')
ii = lambda: int(input())
vi = lambda: list(map(int, input().split()))
vi1 = lambda: list(map(int1, input().split()))


def dbg(*args, **kwargs):
    print(*(repr(arg) for arg in args), *(f'{k}: {repr(v)}' for k, v in kwargs.items()),
          sep='; ', file=sys.stderr, flush=True)


def main():
    n = ii()
    inp = vi1()
    a = [None] * n
    for i, v in enumerate(inp):
        a[v] = i

    p = [None] * n
    for i in range(1, n):
        ptr = i - 1
        if a[i] > a[ptr]:
            p[i] = ptr
            continue

        while p[ptr] is not None and a[i] < a[p[ptr]]:
            ptr = p[ptr]

        p[i] = p[ptr]
        p[ptr] = i

    r = p.index(None)
    adj = [[] for _ in range(n)]
    for i in range(n):
        if p[i] is not None:
            adj[p[i]].append(i)

    h = [-1] * n
    sz = [0] * n

    def dfs(x, d):
        for u in adj[x]:
            dfs(u, d + 1)
            sz[x] += sz[u]

        h[x] = d
        sz[x] += 1

    dfs(r, 0)

    print(*(h[inp[i]] for i in range(n)))
    print(*(sz[inp[i]] - 1 for i in range(n)))


def _start():
    if (ret := main()) is not None:
        print(*ret) if isinstance(ret, List) or isinstance(ret, Tuple) else print(ret)


if __name__ == '__main__':
    _start()
0