結果

問題 No.1094 木登り / Climbing tree
ユーザー takakintakakin
提出日時 2020-07-25 21:26:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,354 ms / 2,000 ms
コード長 1,304 bytes
コンパイル時間 378 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 144,520 KB
最終ジャッジ日時 2024-11-08 07:00:08
合計ジャッジ時間 28,281 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
52,736 KB
testcase_01 AC 1,175 ms
144,268 KB
testcase_02 AC 293 ms
142,244 KB
testcase_03 AC 280 ms
81,528 KB
testcase_04 AC 446 ms
105,408 KB
testcase_05 AC 684 ms
134,840 KB
testcase_06 AC 595 ms
97,756 KB
testcase_07 AC 1,332 ms
143,336 KB
testcase_08 AC 1,299 ms
143,308 KB
testcase_09 AC 1,168 ms
143,696 KB
testcase_10 AC 1,313 ms
143,424 KB
testcase_11 AC 1,345 ms
143,908 KB
testcase_12 AC 1,210 ms
143,660 KB
testcase_13 AC 1,354 ms
143,780 KB
testcase_14 AC 1,301 ms
143,776 KB
testcase_15 AC 432 ms
91,072 KB
testcase_16 AC 645 ms
127,552 KB
testcase_17 AC 553 ms
106,428 KB
testcase_18 AC 507 ms
99,736 KB
testcase_19 AC 604 ms
117,972 KB
testcase_20 AC 1,176 ms
143,280 KB
testcase_21 AC 564 ms
108,880 KB
testcase_22 AC 1,328 ms
143,376 KB
testcase_23 AC 1,175 ms
144,260 KB
testcase_24 AC 1,294 ms
144,520 KB
testcase_25 AC 1,174 ms
144,008 KB
testcase_26 AC 1,199 ms
144,204 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input=lambda: sys.stdin.readline().rstrip()
n=int(input())

Edge=[[] for i in range(n)]
for i in range(n-1):
  a,b,c=map(int,input().split())
  Edge[a-1].append((c,b-1))
  Edge[b-1].append((c,a-1))

inf=float("inf")

prv=[inf]*n #各頂点の親
prv[0]=-1
depth=[inf]*n #深さ
depth[0]=0
stack=[0]
while stack:
  cur=stack.pop()
  for w,next in Edge[cur]:
    if prv[next]==inf:
      prv[next]=cur
      depth[next]=depth[cur]+1
      stack.append(next)

#LCA
LV=(n-1).bit_length() #n=頂点数
def construct(prv):
  kprv=[prv]
  S=prv
  for k in range(LV):
    T=[0]*n
    for i in range(n):
      if S[i] is None:
        continue
      T[i]=S[S[i]]
    kprv.append(T)
    S=T
  return kprv
kprv=construct(prv) # kprv[k][u]=v: 頂点uの2^k個上の祖先頂点v

def lca(u,v,kprv,depth):
  dd=depth[v]-depth[u]
  if dd<0:
    u,v=v,u
    dd=-dd

  for k in range(LV+1):
    if dd&1:
      v=kprv[k][v]
    dd>>=1

  if u==v:
    return u

  for k in range(LV-1,-1,-1):
    pu,pv=kprv[k][u],kprv[k][v]
    if pu!=pv:
      u,v=pu,pv

  return kprv[0][u]

D=[inf]*n
D[0]=0
S=[0]
while S:
	next=S.pop()
	for w,g in Edge[next]:
		if D[g]==inf:
			D[g]=D[next]+w
			S.append(g)

q=int(input())
for _ in range(q):
	s,t=map(int,input().split())
	print(D[s-1]+D[t-1]-2*D[lca(s-1,t-1,kprv,depth)])
0