結果

問題 No.898 tri-βutree
ユーザー lloyzlloyz
提出日時 2022-02-23 22:45:57
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,958 ms / 4,000 ms
コード長 1,781 bytes
コンパイル時間 146 ms
コンパイル使用メモリ 82,332 KB
実行使用メモリ 228,352 KB
最終ジャッジ日時 2024-07-01 23:12:20
合計ジャッジ時間 32,590 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 783 ms
228,352 KB
testcase_01 AC 43 ms
54,016 KB
testcase_02 AC 56 ms
64,128 KB
testcase_03 AC 56 ms
63,744 KB
testcase_04 AC 57 ms
64,128 KB
testcase_05 AC 55 ms
64,128 KB
testcase_06 AC 55 ms
63,744 KB
testcase_07 AC 1,523 ms
129,084 KB
testcase_08 AC 1,532 ms
129,792 KB
testcase_09 AC 1,530 ms
129,920 KB
testcase_10 AC 1,957 ms
129,536 KB
testcase_11 AC 1,542 ms
130,432 KB
testcase_12 AC 1,938 ms
129,768 KB
testcase_13 AC 1,916 ms
129,612 KB
testcase_14 AC 1,928 ms
129,920 KB
testcase_15 AC 1,950 ms
129,536 KB
testcase_16 AC 1,582 ms
129,664 KB
testcase_17 AC 1,519 ms
129,792 KB
testcase_18 AC 1,929 ms
129,164 KB
testcase_19 AC 1,958 ms
130,048 KB
testcase_20 AC 1,934 ms
129,804 KB
testcase_21 AC 1,935 ms
130,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict, deque
import sys
sys.setrecursionlimit(10**6)

n = int(input())

max_log_v = 30

# 親を2^k回辿って到達する頂点(根を通り過ぎる場合は-1とする)
Parent = [[-1 for _ in range(n)] for _ in range(max_log_v)]
Depth = [-1 for _ in range(n)]
def dfs(curr, prev, d):
    Parent[0][curr] = prev
    Depth[curr] = d
    for np, _ in edge[curr]:
        if np == prev:
            continue
        dfs(np, curr, d + 1)

def init():
    root = 0
    # Parent[0]とDepthを根rootの木として初期化する
    dfs(root, -1, 0)
    # Parentを初期化する
    for i in range(max_log_v - 1):
        for v in range(n):
            if Parent[i][v] < 0:
                Parent[i + 1][v] = -1
            else:
                Parent[i + 1][v] = Parent[i][Parent[i][v]]

def check_LCA(u, v):
    if Depth[u] > Depth[v]:
        u, v = v, u
    for i in range(max_log_v):
        if (Depth[v] - Depth[u]) >> i & 1:
            v = Parent[i][v]
    if u == v:
        return u
    for i in range(max_log_v - 1, -1, -1):
        if Parent[i][u] != Parent[i][v]:
            u = Parent[i][u]
            v = Parent[i][v]
    return Parent[0][u]

edge = defaultdict(list)
for _ in range(n - 1):
    u, v, w = map(int, input().split())
    edge[u].append((v, w))
    edge[v].append((u, w))

init()

INF = 1 << 60
D = [INF for _ in range(n)]
D[0] = 0
Que = deque([(0, 0)])
while Que:
    curr, d = Que.popleft()
    for np, w in edge[curr]:
        if d + w > D[np]:
            continue
        D[np] = d + w
        Que.append((np, d + w))

def dist(u, v):
    return D[u] + D[v] - 2 * D[check_LCA(u, v)]

q = int(input())
for _ in range(q):
    x, y, z = map(int, input().split())
    print((dist(x, y) + dist(y, z) + dist(z, x)) // 2)
0