結果
問題 | No.1221 木 *= 3 |
ユーザー |
![]() |
提出日時 | 2020-09-05 10:17:42 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 108 ms / 2,000 ms |
コード長 | 2,062 bytes |
コンパイル時間 | 2,250 ms |
コンパイル使用メモリ | 204,888 KB |
最終ジャッジ日時 | 2025-01-14 07:02:45 |
ジャッジサーバーID (参考情報) |
judge5 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
#include <bits/stdc++.h>#define rep(i, n) for (int i = 0; i < (int)n; ++i)#define rrep(i, n) for (int i = (int)n-1; i >= 0; --i)using namespace std;using ll = long long;template<typename T>inline bool chmax(T& a, const T& b) {if (a < b){a = b;return true;}return false;}template<typename T>inline bool chmin(T& a, const T& b) {if (b < a) {a = b;return true;}return false;}/*** @brief 多次元 vector の作成* @author えびちゃん*/namespace detail {template<typename T, int N>auto make_vec(vector<int>& sizes, T const& x) {if constexpr (N == 1) {return vector(sizes[0], x);} else {int size = sizes[N-1];sizes.pop_back();return vector(size, make_vec<T, N-1>(sizes, x));}}}template<typename T, int N>auto make_vec(int const(&sizes)[N], T const& x = T()) {vector<int> s(N);for (int i = 0; i < N; ++i) s[i] = sizes[N-i-1];return detail::make_vec<T, N>(s, x);}__attribute__((constructor))void fast_io() {ios::sync_with_stdio(false);cin.tie(nullptr);}int main() {int n;cin >> n;vector<int> a(n), b(n);rep(i, n) cin >> a[i];rep(i, n) cin >> b[i];auto g = make_vec<int>({n, 0});rep(i, n-1) {int u, v;cin >> u >> v;u--; v--;g[u].push_back(v);g[v].push_back(u);}const ll inf = 1e18;auto dp = make_vec<ll>({n, 2}, -inf);auto dfs = [&](auto self, int v, int p = -1) -> void {if (g[v].size() == 0) {dp[v][0] = a[v];dp[v][1] = 0;return;}dp[v][0] = 0;dp[v][1] = 0;for (int u : g[v]) {if (u == p) continue;self(self, u, v);dp[v][0] += max(dp[u][0], dp[u][1]);dp[v][1] += max(dp[u][0], dp[u][1] + b[u] + b[v]);}dp[v][0] += a[v];};dfs(dfs, 0);cout << max(dp[0][0], dp[0][1]) << '\n';}