結果
問題 | No.898 tri-βutree |
ユーザー | Navier_Boltzmann |
提出日時 | 2022-01-03 21:25:04 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,249 ms / 4,000 ms |
コード長 | 3,711 bytes |
コンパイル時間 | 268 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 202,880 KB |
最終ジャッジ日時 | 2024-04-21 10:41:48 |
合計ジャッジ時間 | 37,767 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 839 ms
202,880 KB |
testcase_01 | AC | 47 ms
54,656 KB |
testcase_02 | AC | 61 ms
63,232 KB |
testcase_03 | AC | 59 ms
62,464 KB |
testcase_04 | AC | 63 ms
63,872 KB |
testcase_05 | AC | 63 ms
64,128 KB |
testcase_06 | AC | 64 ms
64,128 KB |
testcase_07 | AC | 2,138 ms
124,736 KB |
testcase_08 | AC | 2,236 ms
125,204 KB |
testcase_09 | AC | 2,204 ms
126,092 KB |
testcase_10 | AC | 2,216 ms
124,584 KB |
testcase_11 | AC | 2,210 ms
126,036 KB |
testcase_12 | AC | 2,217 ms
125,512 KB |
testcase_13 | AC | 2,202 ms
126,376 KB |
testcase_14 | AC | 2,249 ms
126,840 KB |
testcase_15 | AC | 2,198 ms
123,796 KB |
testcase_16 | AC | 2,206 ms
123,888 KB |
testcase_17 | AC | 2,201 ms
125,684 KB |
testcase_18 | AC | 2,242 ms
124,780 KB |
testcase_19 | AC | 2,204 ms
125,252 KB |
testcase_20 | AC | 2,249 ms
123,800 KB |
testcase_21 | AC | 2,185 ms
125,688 KB |
ソースコード
import sys sys.setrecursionlimit(10**6) class LCA_wp(): def __init__(self,e,root=0): self.N = len(e) self.e = e self.root = root self.IN = [-1]*N self.OUT = [-1]*N self.DIST = [-1]*N self.time = 0 self.DIST[root] = 0 self.parents = [-1]*N self.dfs(root) self.depth = [(self.DIST[i],i) for i in range(N)] self.init_val = [-1]*(2*N) for i in range(N): x = self.IN[i] y = self.OUT[i] self.init_val[x] = self.depth[i] self.init_val[y] = self.depth[self.parents[i]] self.T = self.SegTree(self.init_val) def dfs(self,x): self.IN[x] = self.time self.time += 1 for ix,c in e[x]: if self.DIST[ix] != -1: continue self.parents[ix] = x self.DIST[ix] = self.DIST[x] + c self.dfs(ix) self.OUT[x] = self.time self.time += 1 def lca(self,l,r): L = self.IN[l] R = self.IN[r] if R<L: L,R = R,L R += 1 return self.T.query(L,R)[1] def distance_between(self,l,r): p = self.lca(l,r) return self.DIST[l]+ self.DIST[r] - 2*self.DIST[p] def distance_from_root(self,x): return self.DIST[x] class SegTree: """ Segment Tree """ def __init__(self, init_val, segfunc = None, ide_ele= None): """ 初期化 init_val: 配列の初期値 """ if segfunc is None: def segfunc(x,y): return min(x,y) ide_ele = (float('inf'),-1) n = len(init_val) self.segfunc = segfunc self.ide_ele = ide_ele self.num = 1 << (n - 1).bit_length() self.tree = [ide_ele] * 2 * self.num # 配列の値を葉にセット for i in range(n): self.tree[self.num + i] = init_val[i] # 構築していく for i in range(self.num - 1, 0, -1): self.tree[i] = segfunc(self.tree[2 * i], self.tree[2 * i + 1]) def update(self, k, x): """ k番目の値をxに更新 k: index(0-index) x: update value """ k += self.num self.tree[k] = x while k > 1: self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1]) k >>= 1 def query(self, l, r): """ [l, r)のsegfuncしたものを得る l: index(0-index) r: index(0-index) """ res = self.ide_ele l += self.num r += self.num while l < r: if l & 1: res = self.segfunc(res, self.tree[l]) l += 1 if r & 1: res = self.segfunc(res, self.tree[r - 1]) l >>= 1 r >>= 1 return res N = int(input()) e = [[] for _ in range(N)] from collections import defaultdict dist = defaultdict(lambda:0) for _ in range(N-1): a,b,c = map(int,input().split()) e[a].append((b,c)) e[b].append((a,c)) L = LCA_wp(e) Q = int(input()) for _ in range(Q): x,y,z = map(int,input().split()) X = L.distance_between(x,y) Y = L.distance_between(z,y) Z = L.distance_between(x,z) print((X+Y+Z)//2)