結果

問題 No.898 tri-βutree
コンテスト
ユーザー terasa
提出日時 2022-05-31 10:55:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,091 ms / 4,000 ms
コード長 2,651 bytes
コンパイル時間 276 ms
コンパイル使用メモリ 82,340 KB
実行使用メモリ 206,208 KB
最終ジャッジ日時 2024-09-21 01:12:07
合計ジャッジ時間 22,140 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import pypyjit
import itertools
import heapq
import math
from collections import deque, defaultdict
import bisect

input = sys.stdin.readline
sys.setrecursionlimit(10 ** 6)
pypyjit.set_param('max_unroll_recursion=-1')


def index_lt(a, x):
    'return largest index s.t. A[i] < x or -1 if it does not exist'
    return bisect.bisect_left(a, x) - 1


def index_le(a, x):
    'return largest index s.t. A[i] <= x or -1 if it does not exist'
    return bisect.bisect_right(a, x) - 1


def index_gt(a, x):
    'return smallest index s.t. A[i] > x or len(a) if it does not exist'
    return bisect.bisect_right(a, x)


def index_ge(a, x):
    'return smallest index s.t. A[i] >= x or len(a) if it does not exist'
    return bisect.bisect_left(a, x)


class LCA:
    def __init__(self, N, E, root=0):
        self.N = N
        self.E = E
        self.K = N.bit_length()
        self.par = [[-1 for _ in range(N)] for _ in range(self.K)]
        self.depth = [None] * N

        self.order = [None] * N
        self._cnt = 0

        self.cost = [0] * N

        self.dfs(root, 0, -1)
        for k in range(self.K - 1):
            for i in range(self.N):
                if self.par[k][i] < 0:
                    continue
                self.par[k + 1][i] = self.par[k][self.par[k][i]]

    def dfs(self, v, d, p):
        self.par[0][v] = p
        self.depth[v] = d
        self.order[v] = self._cnt
        self._cnt += 1

        for c, dest in self.E[v]:
            if dest == p:
                continue
            self.cost[dest] = self.cost[v] + c
            self.dfs(dest, d + 1, v)

    def lca(self, u, v):
        if self.depth[u] > self.depth[v]:
            u, v = v, u
        d = self.depth[v] - self.depth[u]
        for k in range(self.K):
            if d & (1 << k):
                v = self.par[k][v]

        if u == v:
            return u
        for k in range(self.K)[::-1]:
            if self.par[k][u] != self.par[k][v]:
                u = self.par[k][u]
                v = self.par[k][v]
        return self.par[0][v]

    def dist(self, u, v):
        return self.cost[u] + self.cost[v] - 2 * self.cost[self.lca(u, v)]


N = int(input())
E = [[] for _ in range(N)]
for _ in range(N - 1):
    u, v, w = map(int, input().split())
    E[u].append((w, v))
    E[v].append((w, u))
solver = LCA(N, E)

Q = int(input())
for _ in range(Q):
    x, y, z = map(int, input().split())
    xo, yo, zo = solver.order[x], solver.order[y], solver.order[z]
    x, y, z = map(lambda x: x[0], sorted([(x, xo), (y, yo), (z, zo)], key=lambda x: x[1]))
    print((solver.dist(x, y) + solver.dist(y, z) + solver.dist(z, x)) // 2)
0