結果

問題 No.898 tri-βutree
コンテスト
ユーザー yaneko
提出日時 2025-11-19 15:48:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,238 ms / 4,000 ms
コード長 2,480 bytes
コンパイル時間 1,482 ms
コンパイル使用メモリ 81,784 KB
実行使用メモリ 115,920 KB
最終ジャッジ日時 2025-11-19 17:11:27
合計ジャッジ時間 24,508 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

class LCA:
    def __init__(self, graph, root=0):
        self.tree = graph
        self.root = root
        self.n = len(graph)
        self.k = self._get_bit()
        self.dist = [-1]*self.n
        self.weight = [0]*self.n
        self.parent = [[-1]*self.n for _ in range(self.k)]
        self._dfs()
        self._doubling()

    def _get_bit(self):
        k = 1
        while (1<<k)<self.n: k += 1
        return k
    
    def _dfs(self):
        self.dist[self.root] = 0
        stack = [self.root]
        while stack:
            u = stack.pop()
            for v, w in self.tree[u]:
                if self.dist[v]!=-1:
                    continue
                self.dist[v] = self.dist[u]+1
                self.weight[v] = self.weight[u]+w
                self.parent[0][v] = u
                stack.append(v)

    def _doubling(self):
        for i in range(self.k-1):
            for j in range(self.n):
                if self.parent[i][j]<0:
                    self.parent[i+1][j] = -1
                else:
                    self.parent[i+1][j] = self.parent[i][self.parent[i][j]]

    def query(self, u, v):
        if self.dist[u]<self.dist[v]:
            u, v = v, u
        for i in range(self.k):
            if (self.dist[u]-self.dist[v])>>i&1:
                u = self.parent[i][u]
        if u==v:
            return u
        for i in reversed(range(self.k)):
            if self.parent[i][u]!=self.parent[i][v]:
                u = self.parent[i][u]
                v = self.parent[i][v]
        return self.parent[0][u]
    
    def get_dist(self, u, v):
        return self.dist[u]+self.dist[v]-2*self.dist[self.query(u, v)]

    def get_weight(self, u, v):
        return self.weight[u]+self.weight[v]-2*self.weight[self.query(u, v)]
    
    def is_on_path(self, u, v, a):
        return self.get_dist(u, a)+self.get_dist(a, v)==self.get_dist(u, v)
    
    def kth_ancestor(self, u, k):
        for i in range(self.k):
            if (k>>i)&1:
                u = self.parent[i][u]
                if u == -1:
                    break
        return u
    
if __name__=='__main__':
    n = int(input())
    t = [[] for _ in range(n)]
    for _ in range(n-1):
        u, v, w = map(int, input().split())
        t[u].append((v, w))
        t[v].append((u, w))
    lca = LCA(t)
    for _ in range(int(input())):
        x, y, z = map(int, input().split())
        print((lca.get_weight(x, y)+lca.get_weight(y, z)+lca.get_weight(z, x))//2)
0