結果

問題 No.1221 木 *= 3
ユーザー donutholedonuthole
提出日時 2020-09-05 15:10:09
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 2,208 bytes
コンパイル時間 304 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 55,084 KB
最終ジャッジ日時 2024-11-28 16:01:26
合計ジャッジ時間 14,435 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 13 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

import sys
import math
import collections
import bisect
import itertools
import decimal
import copy
from math import log2
# import numpy as np
# sys.setrecursionlimit(10 ** 6)
INF = 10 ** 20
# MOD = 10 ** 9 + 7
# MOD = 998244353
ni = lambda: int(sys.stdin.readline().rstrip())
ns = lambda: map(int, sys.stdin.readline().rstrip().split())
na = lambda: list(map(int, sys.stdin.readline().rstrip().split()))
na1 = lambda: list(map(lambda x: int(x) - 1, sys.stdin.readline().rstrip().split()))
# ===CODE===
def main():
n = ni()
a = na()
b = na()
e = [[] for _ in range(n)]
for _ in range(n - 1):
u, v = na1()
e[u].append(v)
e[v].append(u)
# dp[v][x]:Node vx
# x=0:Node v使使
# x=1:Node v使
# x=2:Node v使
# x=3:Node v 32
dp = [[0 for _ in range(4)] for __ in range(n)]
def dfs(node, parent):
dp[node][0] = b[node] # node使 node==0b[node]使OK
dp[node][1] = 0 #
dp[node][2] = a[node] # node使 & node使
dp[node][3] = a[node] # node使
for ei in e[node]:
if ei != parent:
dfs(ei, node)
dp[node][0] += max(dp[ei][0] + b[node], dp[ei][2])
dp[node][1] += max(dp[ei][0] + b[node], dp[ei][2])
dp[node][2] += max(dp[ei][1], dp[ei][3])
dp[node][3] += max(dp[ei][1], dp[ei][3])
dfs(0, 0)
# Node 013
ans = max(dp[0][1], dp[0][3])
print(ans)
if __name__ == '__main__':
main()
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0