結果
問題 |
No.1221 木 *= 3
|
ユーザー |
|
提出日時 | 2025-07-24 17:07:35 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 125 ms / 2,000 ms |
コード長 | 2,347 bytes |
コンパイル時間 | 3,694 ms |
コンパイル使用メモリ | 289,308 KB |
実行使用メモリ | 29,312 KB |
最終ジャッジ日時 | 2025-07-24 17:07:44 |
合計ジャッジ時間 | 8,262 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
#include <bits/stdc++.h> using namespace std; template <class F> class y_combinator { F f; public: y_combinator(F&& f) : f(std::forward<F>(f)) {} template <class... Args> auto operator()(Args&&... args) const { return f(*this, std::forward<Args>(args)...); } }; using ll = long long; using ld = long double; template <class T, class U = std::less<T>> using prique = std::priority_queue<T, std::vector<T>, U>; template <class T> T floor(T a, T b) noexcept { return a / b - (a % b && (a ^ b) < 0); } template <class T> T ceil(T a, T b) noexcept { return floor(a + b - 1, b); } template <class T> bool chmin(T& x, const T& y) noexcept { return (x > y ? x = y, true : false); } template <class T> bool chmax(T& x, const T& y) noexcept { return (x < y ? x = y, true : false); } #define overload4(a, b, c, d, e, ...) e #define rep1(a) for (long long _i = 0; _i < (a); _i++) #define rep2(i, a) for (long long i = 0; i < (a); i++) #define rep3(i, a, b) for (long long i = (a); i < (b); i++) #define rep4(i, a, b, c) for (long long i = (a); i < (b); i += (c)) #define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__) #define rrep(i, a, b, c) for (long long i = (a); i > (b); i += (c)) #define all(x) std::begin(x), std::end(x) #define rall(x) std::rbegin(x), std::rend(x) #define pb push_back #ifndef LOCAL #define debug(...) #endif void run_case() { int N; cin >> N; vector<int> A(N), B(N); rep(i, N) cin >> A[i]; rep(i, N) cin >> B[i]; vector<vector<int>> G(N); rep(i, 1, N) { int a, b; cin >> a >> b; a--, b--; G[a].pb(b); G[b].pb(a); } vector dp(N, vector<ll>(2, 0)); y_combinator([&](auto self, int v, int p = -1) -> void { vector<pair<ll, ll>> C; for (int u : G[v]) if (u != p) { self(u, v); C.pb({dp[u][1], u}); dp[v][0] += max(dp[u][0], dp[u][1]); } dp[v][0] += A[v]; for(int u: G[v]) if(u != p) { dp[v][1] += max(dp[u][1] + B[u] + B[v], dp[u][0]); } })(0); debug(dp); cout << max(dp[0][0], dp[0][1]) << "\n"; } int main() { std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); std::fixed(std::cout).precision(16); int T = 1; while (T--) run_case(); return 0; }