結果
問題 | No.1221 木 *= 3 |
ユーザー |
![]() |
提出日時 | 2020-09-25 05:23:54 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 190 ms / 2,000 ms |
コード長 | 1,610 bytes |
コンパイル時間 | 1,594 ms |
コンパイル使用メモリ | 169,636 KB |
実行使用メモリ | 21,248 KB |
最終ジャッジ日時 | 2024-06-28 05:42:55 |
合計ジャッジ時間 | 6,749 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
#include<bits/stdc++.h>using namespace std;#define fs first#define sc second#define pb push_back#define mp make_pair#define eb emplace_back#define ALL(A) A.begin(),A.end()#define RALL(A) A.rbegin(),A.rend()typedef long long ll;typedef pair<ll,ll> P;template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }template<typename T> T gcd(T a,T b){return b?gcd(b,a%b):a;}const ll mod=998244353;const ll LINF=1ll<<60;const int INF=1<<30;int dx[]={1,0,-1,0,1,-1,1,-1};int dy[]={0,1,0,-1,1,-1,-1,1};ll a[1<<17], b[1<<17];ll dp[1<<17][2];vector<int> G[1<<17];int n;ll dfs(int u, int v, bool f){if(dp[u][f] != -LINF) return dp[u][f];ll ret = 0;if(f){for(auto &g:G[u]){if(g == v) continue;ret += max(dfs(g, u, 1) + b[u] + b[g], dfs(g, u, 0));}}else{for(auto &g:G[u]){if(g == v) continue;ret += max(dfs(g, u, 1), dfs(g, u, 0));}ret += a[u];}return dp[u][f] = ret;}int main(){cin >> n;for (int i = 0; i < n; i++) {cin >> a[i];}for (int i = 0; i < n; i++) {cin >> b[i];}for (int i = 0; i < n-1; i++) {int u,v;cin >> u >> v;u--,v--;G[u].pb(v);G[v].pb(u);}for (int i = 0; i < 1<<17; i++) {for (int j = 0; j < 2; j++) {dp[i][j] = -LINF;}}cout << max(dfs(0, 0, 0), dfs(0, 0, 1)) << endl;return 0;}