結果
問題 | No.1221 木 *= 3 |
ユーザー |
|
提出日時 | 2021-01-01 16:04:59 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 87 ms / 2,000 ms |
コード長 | 2,128 bytes |
コンパイル時間 | 1,441 ms |
コンパイル使用メモリ | 139,664 KB |
最終ジャッジ日時 | 2025-01-17 08:54:02 |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
#include <iostream>#include <vector>#include <algorithm>#include <cmath>#include <queue>#include <string>#include <map>#include <set>#include <stack>#include <tuple>#include <deque>#include <array>#include <numeric>#include <bitset>#include <iomanip>#include <cassert>#include <chrono>#include <random>#include <limits>#include <iterator>#include <functional>#include <sstream>#include <fstream>#include <complex>#include <cstring>#include <unordered_map>#include <unordered_set>using namespace std;using ll = long long;using P = pair<int, int>;constexpr int INF = 1001001001;constexpr int mod = 1000000007;// constexpr int mod = 998244353;template<class T>inline bool chmax(T& x, T y){if(x < y){x = y;return true;}return false;}template<class T>inline bool chmin(T& x, T y){if(x > y){x = y;return true;}return false;}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int n;cin >> n;vector<int> a(n), b(n);for(int i = 0; i < n; ++i) cin >> a[i];for(int i = 0; i < n; ++i) cin >> b[i];vector<vector<int>> graph(n);for(int i = 1; i < n; ++i){int u, v;cin >> u >> v;--u, --v;graph[u].emplace_back(v);graph[v].emplace_back(u);}constexpr ll inf = 1e+17;vector<vector<ll>> dp(n, vector<ll>(2, -inf));auto func = [&](auto&& self, int v = 0, int i = 0, int par = -1) -> ll {if(dp[v][i] != -inf) return dp[v][i];ll res = 0;if(i == 0){res = a[v];for(int child : graph[v]){if(child == par) continue;res += max(self(self, child, 0, v), self(self, child, 1, v));}}else{for(int child : graph[v]){if(child == par) continue;res += max(self(self, child, 0, v), self(self, child, 1, v) + b[v] + b[child]);}}return dp[v][i] = res;};cout << max(func(func), func(func, 0, 1)) << endl;return 0;}