結果

問題 No.1221 木 *= 3
ユーザー persimmon-persimmonpersimmon-persimmon
提出日時 2021-03-19 14:57:21
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,280 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 82,380 KB
実行使用メモリ 123,884 KB
最終ジャッジ日時 2024-11-18 09:33:13
合計ジャッジ時間 5,877 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
52,988 KB
testcase_01 AC 39 ms
53,940 KB
testcase_02 AC 36 ms
52,264 KB
testcase_03 AC 35 ms
53,548 KB
testcase_04 WA -
testcase_05 AC 36 ms
54,348 KB
testcase_06 AC 33 ms
53,920 KB
testcase_07 AC 192 ms
112,344 KB
testcase_08 AC 202 ms
114,544 KB
testcase_09 AC 239 ms
121,640 KB
testcase_10 AC 243 ms
122,780 KB
testcase_11 AC 252 ms
122,936 KB
testcase_12 AC 203 ms
123,884 KB
testcase_13 AC 198 ms
122,968 KB
testcase_14 AC 213 ms
123,400 KB
testcase_15 AC 208 ms
123,448 KB
testcase_16 AC 210 ms
123,456 KB
testcase_17 AC 242 ms
121,680 KB
testcase_18 AC 248 ms
120,028 KB
testcase_19 AC 251 ms
118,172 KB
testcase_20 AC 251 ms
120,128 KB
testcase_21 AC 250 ms
116,776 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main1(n,a,b,uv):
  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,子を消すときの最大,子を消さないときの最大]の配列
  dp1=[0]*n
  # dp1[v]:頂点vを根とした部分木に対する解
  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