結果

問題 No.3134 二分探索木
コンテスト
ユーザー flippergo
提出日時 2026-04-11 10:42:40
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 1,093 ms / 2,000 ms
コード長 772 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 225 ms
コンパイル使用メモリ 85,632 KB
実行使用メモリ 411,224 KB
最終ジャッジ日時 2026-04-11 10:42:51
合計ジャッジ時間 9,649 ms
ジャッジサーバーID
（参考情報） 		judge3_1 / judge1_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
sys.setrecursionlimit(10**6)
N = int(input())
A = list(map(int, input().split()))
pos = [0]*(N+1)
for i in range(N):
    pos[A[i]] = i
G = {i:[-1,-1] for i in range(1,N+1)}
stack = []
for i in range(1,N+1):
    last = -1
    while stack and pos[i]<pos[stack[-1]]:
        last = stack.pop()
    if last>0:
        G[i][0] = last
    if len(stack)>0:
        G[stack[-1]][1] = i
    stack.append(i)
B = [-1]*N
B[pos[A[0]]] = 0
val = [[0,0] for _ in range(N)]
cnt = 0
def dfs(v):
    global cnt
    for u in G[v]:
        if u<0:continue
        B[pos[u]] = B[pos[v]]+1
        cnt += 1
        val[pos[u]][0] = cnt
        dfs(u)
    cnt += 1
    val[pos[v]][1] = cnt
dfs(A[0])
C = [0]*N
for i in range(N):
    C[i] = (val[i][1]-val[i][0])//2
print(*B)
print(*C)
0