結果
問題 | No.1221 木 *= 3 |
ユーザー |
![]() |
提出日時 | 2020-11-22 17:20:01 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 214 ms / 2,000 ms |
コード長 | 1,680 bytes |
コンパイル時間 | 1,867 ms |
コンパイル使用メモリ | 177,508 KB |
実行使用メモリ | 30,208 KB |
最終ジャッジ日時 | 2024-07-23 16:39:24 |
合計ジャッジ時間 | 7,301 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
#include <bits/stdc++.h>using namespace std;#define rep(i, n) for(int(i) = 0; (i) < (n); (i)++)#define FOR(i, m, n) for(int(i) = (m); (i) < (n); (i)++)#define All(v) (v).begin(), (v).end()#define pb push_back#define MP(a, b) make_pair((a), (b))template <class T> vector<T> make_vec(size_t a, T val) {return vector<T>(a, val);}template <class... Ts> auto make_vec(size_t a, Ts... ts) {return vector<decltype(make_vec(ts...))>(a, make_vec(ts...));}using ll = long long;using pii = pair<int, int>;using pll = pair<ll, ll>;using Graph = vector<vector<int>>;template <typename T> struct edge {int to;T cost;edge(int t, T c) : to(t), cost(c) {}};template <typename T> using WGraph = vector<vector<edge<T>>>;const int INF = 1 << 30;const ll LINF = 1LL << 60;const int MOD = 1e9 + 7;int main() {int N;cin >> N;vector<ll> A(N), B(N);rep(i, N) { cin >> A[i]; }rep(i, N) { cin >> B[i]; }Graph G(N);rep(i, N - 1) {int u, v;cin >> u >> v;u--, v--;G[u].pb(v);G[v].pb(u);}auto dp = make_vec(N, 2, -LINF);function<ll(int, int, int)> dfs = [&](int v, int p, int s) {if(dp[v][s] != -LINF)return dp[v][s];ll ret = 0;for(auto c : G[v]) {if(c == p)continue;if(s == 0) {ret += max(dfs(c, v, 0), dfs(c, v, 1));} else {ret += max(dfs(c, v, 0), dfs(c, v, 1) + B[c] + B[v]);}}if(!s)ret += A[v];return dp[v][s] = ret;};cout << max(dfs(0, -1, 0), dfs(0, -1, 1)) << endl;}