結果
問題 | No.1221 木 *= 3 |
ユーザー |
![]() |
提出日時 | 2020-07-18 16:45:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 374 ms / 2,000 ms |
コード長 | 733 bytes |
コンパイル時間 | 237 ms |
コンパイル使用メモリ | 82,324 KB |
実行使用メモリ | 111,872 KB |
最終ジャッジ日時 | 2024-11-15 16:16:34 |
合計ジャッジ時間 | 7,190 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
import sys def input(): return sys.stdin.buffer.readline()[:-1] n = int(input()) a = list(map(int, input().split())) b = list(map(int, input().split())) adj = [[] for _ in range(n)] for i in range(n-1): u, v = map(int, input().split()) adj[u-1].append((v-1, b[u-1] + b[v-1])) adj[v-1].append((u-1, b[u-1] + b[v-1])) dp = [[0 for _ in range(2)] for _ in range(n)] visited = [False for _ in range(n)] stack = [(0, -1)] while stack: i, p = stack[-1] if not visited[i]: visited[i] = True for j, e in adj[i]: if j != p: stack.append((j, i)) else: stack.pop() dp[i][0] = a[i] for j, e in adj[i]: if j != p: dp[i][0] += max(dp[j][0], dp[j][1]) dp[i][1] += max(dp[j][0], dp[j][1] + e) print(max(dp[0]))