結果

問題 No.1221 木 *= 3
ユーザー persimmon-persimmonpersimmon-persimmon
提出日時 2021-03-19 15:03:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 327 ms / 2,000 ms
コード長 1,245 bytes
コンパイル時間 218 ms
コンパイル使用メモリ 82,576 KB
実行使用メモリ 123,940 KB
最終ジャッジ日時 2024-11-18 09:38:31
合計ジャッジ時間 6,227 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
52,688 KB
testcase_01 AC 40 ms
53,656 KB
testcase_02 AC 39 ms
52,836 KB
testcase_03 AC 40 ms
52,624 KB
testcase_04 AC 39 ms
53,372 KB
testcase_05 AC 39 ms
52,744 KB
testcase_06 AC 40 ms
52,768 KB
testcase_07 AC 246 ms
112,428 KB
testcase_08 AC 257 ms
114,264 KB
testcase_09 AC 300 ms
120,816 KB
testcase_10 AC 311 ms
121,624 KB
testcase_11 AC 311 ms
121,532 KB
testcase_12 AC 255 ms
123,496 KB
testcase_13 AC 261 ms
122,860 KB
testcase_14 AC 278 ms
123,940 KB
testcase_15 AC 261 ms
123,552 KB
testcase_16 AC 269 ms
123,564 KB
testcase_17 AC 307 ms
121,012 KB
testcase_18 AC 320 ms
119,636 KB
testcase_19 AC 322 ms
117,400 KB
testcase_20 AC 315 ms
119,468 KB
testcase_21 AC 327 ms
115,760 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main1(n,a,b,uv):
  if n==1:
    return max(a[0],0)
  ki=[[] for _ in range(n)]
  for u,v in uv:
    u,v=u-1,v-1
    ki[u].append(v)
    ki[v].append(u)
  # 頂点vを消したときの最大値
  # 頂点vを消さないときの最大値
  tree_order=[]
  parent=[-1]*n
  todo=[[0,-1]]
  while todo:
    v,p=todo.pop()
    tree_order.append(v)
    parent[v]=p
    for nv in ki[v]:
      if nv==p:continue
      todo.append([nv,v])
  tree_order.reverse()
  ary=[[] for _ in range(n)]
  # ary[v]:頂点vが子から遷移してきた値[子のidx,子を消すときの最大,子を消さないときの最大]の配列
  ans=0
  for v in tree_order:
    p=parent[v]
    if len(ary[v])==0:
      ary[p].append([v,a[v],0])
    else:
      num1=a[v] # 頂点vを消す
      num2=0 # 頂点vを消さない
      for i,ai,bi in ary[v]:
        num1+=max(ai,bi)
        num2+=max(ai,bi+b[i]+b[v])
      if p>=0:
        ary[p].append([v,num1,num2])
      else:
        ans=max(num1,num2)
  return ans

import sys
input=sys.stdin.readline
if __name__=='__main__':
  n=int(input())
  a=list(map(int,input().split()))
  b=list(map(int,input().split()))
  uv=[list(map(int,input().split())) for _ in range(n-1)]
  ret1=main1(n,a,b,uv)
  print(ret1)
0