結果
問題 | No.898 tri-βutree |
ユーザー | kurimupy |
提出日時 | 2020-06-18 23:08:03 |
言語 | PyPy3 (7.3.13) |
結果 |
AC
|
実行時間 | 1,186 ms / 4,000 ms |
コード長 | 2,432 bytes |
コンパイル時間 | 973 ms |
コンパイル使用メモリ | 87,160 KB |
実行使用メモリ | 223,552 KB |
最終ジャッジ日時 | 2023-08-08 16:51:33 |
合計ジャッジ時間 | 22,701 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 658 ms
223,552 KB |
testcase_01 | AC | 71 ms
71,144 KB |
testcase_02 | AC | 80 ms
76,544 KB |
testcase_03 | AC | 80 ms
76,160 KB |
testcase_04 | AC | 79 ms
76,240 KB |
testcase_05 | AC | 80 ms
76,344 KB |
testcase_06 | AC | 81 ms
76,440 KB |
testcase_07 | AC | 1,180 ms
130,620 KB |
testcase_08 | AC | 1,132 ms
130,164 KB |
testcase_09 | AC | 1,138 ms
130,984 KB |
testcase_10 | AC | 1,173 ms
130,804 KB |
testcase_11 | AC | 1,151 ms
132,996 KB |
testcase_12 | AC | 1,132 ms
130,564 KB |
testcase_13 | AC | 1,130 ms
129,252 KB |
testcase_14 | AC | 1,138 ms
130,864 KB |
testcase_15 | AC | 1,144 ms
129,476 KB |
testcase_16 | AC | 1,186 ms
131,300 KB |
testcase_17 | AC | 1,143 ms
132,124 KB |
testcase_18 | AC | 1,149 ms
132,292 KB |
testcase_19 | AC | 1,164 ms
129,492 KB |
testcase_20 | AC | 1,169 ms
131,396 KB |
testcase_21 | AC | 1,154 ms
130,764 KB |
ソースコード
#lca verify import sys sys.setrecursionlimit(10**7) class LCA_Tree(): # 頂点は0スタートでも1スタートでもいい # 頂点の個数だけは記録しておく。 セグメント木の実装verをしてもいいかもしれないけど、こっちのが十分早そうw def bfs(self, node): # 初期化補助関数 self.visited[node] = True for (p,w) in tree[node]: if self.visited[p] == False: self.kprv[0][p] = node self.depth[p] = self.depth[node]+1 self.real_depth[p]=self.real_depth[node]+w self.bfs(p) def __init__(self, root, tree): self.root = root self.n = len(tree)-1 self.K = (self.n-1).bit_length() self.visited = [False for i in range(self.n+1)] self.depth = [0 for i in range(self.n+1)] self.real_depth=[0 for i in range(self.n+1)] self.kprv = [[-1 for i in range(self.n+1)] for j in range(self.K+1)] self.bfs(self.root) for cnt in range(1,self.K+1): for node in range(self.n+1): if self.kprv[cnt-1][node] >= 0: self.kprv[cnt][node] = self.kprv[cnt - 1][self.kprv[cnt-1][node]] else: self.kprv[cnt][node] = -1 def query(self, u, v): # u,vの最小共通祖先 dd = self.depth[v]-self.depth[u] if dd < 0: u, v = v, u dd = -dd for i in range(self.K+1): if dd & 1: v = self.kprv[i][v] dd >>= 1 if u == v: return u for i in range(self.K-1, -1, -1): pu = self.kprv[i][u] pv = self.kprv[i][v] if pu != pv: u = pu v = pv return self.kprv[0][u] N = int(input()) tree = [[] for i in range(N)] for i in range(N-1): u, v, w = map(int, input().split()) tree[u].append((v, w)) tree[v].append((u, w)) lca = LCA_Tree(0, tree) Q = int(input()) que = [tuple(map(int, input().split())) for i in range(Q)] for a, b, c in que: x=lca.query(a, b) y=lca.query(b, c) z=lca.query(c, a) ans = 0 ans += lca.real_depth[a]+lca.real_depth[b]-2*lca.real_depth[x] ans += lca.real_depth[b]+lca.real_depth[c]-2*lca.real_depth[y] ans += lca.real_depth[c]+lca.real_depth[a]-2*lca.real_depth[z] print(ans//2)