結果

問題 No.898 tri-βutree
ユーザー hedwig100
提出日時 2020-04-14 10:58:44
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 2,529 ms / 4,000 ms
コード長 2,036 bytes
コンパイル時間 561 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 83,968 KB
最終ジャッジ日時 2024-11-08 23:40:43
合計ジャッジ時間 43,168 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
sys.setrecursionlimit(100000000)
dy = (-1,0,1,0)
dx = (0,1,0,-1)

class LCA():

    def __init__(self,G,root = 0): #O(NlogN)
        self.n = len(G)
        self.G = G
        self.root = root
        self.K = 1 #2 ** K > nとなる最小のK
        while (1<<self.K) <= self.n:
            self.K += 1
        self.parents = [[-1] * self.n for _ in range(self.K)]
        self.dist = [0] * self.n
        self.depth = [0] * self.n
        
        self._dfs(root,0,0,-1)
        for k in range(self.K - 1):
            for v in range(self.n):
                if self.parents[k][v] >= 0:
                    self.parents[k + 1][v] = self.parents[k][self.parents[k][v]]
    
    def _dfs(self,i,d,d2,p):
        self.parents[0][i] = p
        self.dist[i] = d  
        self.depth[i] = d2

        for e,c in self.G[i]:
            if e != p:
                self._dfs(e,d + 1,d2 + c,i)
    
    def qry(self,u,v): #O(logN)
        if self.dist[u] < self.dist[v]:
            u,v = v,u
        s,t = u,v
        
        dif = self.dist[u] - self.dist[v]
        for k in range(self.K):
            if dif >> k & 1:
                u = self.parents[k][u]
        
        if u == v:
            return self.depth[s] + self.depth[t] - 2 * self.depth[u]
        
        for k in range(self.K - 1,-1,-1):
            if self.parents[k][u] != self.parents[k][v]:
                u = self.parents[k][u]
                v = self.parents[k][v]
        u = self.parents[0][u]
        
        return self.depth[s] + self.depth[t] - 2 * self.depth[u]

def main():
    n = int(input())
    G = [[] for _ in range(n)]
    for _ in range(n - 1):
        a,b,c = map(int,input().split())
        G[a].append((b,c))
        G[b].append((a,c))
    
    lca = LCA(G)
    q = int(input())
    for _ in range(q):
        x,y,z = map(int,input().split())
        ans = 0
        ans += lca.qry(x,y)
        ans += lca.qry(y,z)
        ans += lca.qry(z,x)
        print(ans//2)

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