結果
問題 | No.1221 木 *= 3 |
ユーザー | anagohirame |
提出日時 | 2020-07-18 16:44:11 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 918 ms / 2,000 ms |
コード長 | 628 bytes |
コンパイル時間 | 312 ms |
コンパイル使用メモリ | 10,944 KB |
実行使用メモリ | 81,108 KB |
最終ジャッジ日時 | 2023-08-09 19:18:02 |
合計ジャッジ時間 | 14,028 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 16 ms
7,724 KB |
testcase_01 | AC | 16 ms
7,808 KB |
testcase_02 | AC | 16 ms
7,872 KB |
testcase_03 | AC | 16 ms
7,776 KB |
testcase_04 | AC | 15 ms
7,812 KB |
testcase_05 | AC | 16 ms
7,848 KB |
testcase_06 | AC | 16 ms
7,732 KB |
testcase_07 | AC | 871 ms
79,888 KB |
testcase_08 | AC | 918 ms
81,108 KB |
testcase_09 | AC | 892 ms
79,904 KB |
testcase_10 | AC | 884 ms
81,060 KB |
testcase_11 | AC | 878 ms
81,068 KB |
testcase_12 | AC | 699 ms
62,104 KB |
testcase_13 | AC | 696 ms
62,152 KB |
testcase_14 | AC | 688 ms
62,148 KB |
testcase_15 | AC | 702 ms
62,196 KB |
testcase_16 | AC | 694 ms
62,132 KB |
testcase_17 | AC | 817 ms
65,036 KB |
testcase_18 | AC | 806 ms
65,992 KB |
testcase_19 | AC | 824 ms
65,416 KB |
testcase_20 | AC | 811 ms
65,416 KB |
testcase_21 | AC | 812 ms
65,480 KB |
ソースコード
import sys def input(): return sys.stdin.buffer.readline()[:-1] sys.setrecursionlimit(10**6) 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)] def dfs(i, p): dp[i][0] = a[i] for j, e in adj[i]: if j == p: continue dfs(j, i) dp[i][0] += max(dp[j][0], dp[j][1]) dp[i][1] += max(dp[j][0], dp[j][1] + e) return dfs(0, -1) #print(*dp, sep="\n") print(max(dp[0]))