結果
問題 | No.1221 木 *= 3 |
ユーザー | theory_and_me |
提出日時 | 2020-09-04 23:24:16 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 105 ms / 2,000 ms |
コード長 | 3,152 bytes |
コンパイル時間 | 2,163 ms |
コンパイル使用メモリ | 205,440 KB |
実行使用メモリ | 31,060 KB |
最終ジャッジ日時 | 2023-08-17 20:46:33 |
合計ジャッジ時間 | 5,040 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 2 ms
4,376 KB |
testcase_02 | AC | 1 ms
4,380 KB |
testcase_03 | AC | 2 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,380 KB |
testcase_05 | AC | 1 ms
4,380 KB |
testcase_06 | AC | 1 ms
4,376 KB |
testcase_07 | AC | 99 ms
31,060 KB |
testcase_08 | AC | 99 ms
30,244 KB |
testcase_09 | AC | 102 ms
31,040 KB |
testcase_10 | AC | 105 ms
30,428 KB |
testcase_11 | AC | 105 ms
30,500 KB |
testcase_12 | AC | 75 ms
14,036 KB |
testcase_13 | AC | 75 ms
13,824 KB |
testcase_14 | AC | 73 ms
13,968 KB |
testcase_15 | AC | 76 ms
13,888 KB |
testcase_16 | AC | 75 ms
14,012 KB |
testcase_17 | AC | 99 ms
10,680 KB |
testcase_18 | AC | 99 ms
10,696 KB |
testcase_19 | AC | 103 ms
10,712 KB |
testcase_20 | AC | 101 ms
10,644 KB |
testcase_21 | AC | 100 ms
10,692 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; #define REP(i,n) for(ll i=0;i<(ll)n;i++) #define dump(x) cerr << "Line " << __LINE__ << ": \n" << #x << " = " << (x) << "\n"; #define spa << " " << #define fi first #define se second #define ALL(a) (a).begin(),(a).end() #define ALLR(a) (a).rbegin(),(a).rend() using ld = long double; using ll = long long; using ull = unsigned long long; using pii = pair<int, int>; using pll = pair<long, long>; using pdd = pair<ld, ld>; template<typename T> using V = vector<T>; template<typename T> using P = pair<T, T>; template<typename T> vector<T> make_vec(size_t n, T a) { return vector<T>(n, a); } template<typename... Ts> auto make_vec(size_t n, Ts... ts) { return vector<decltype(make_vec(ts...))>(n, make_vec(ts...)); } template<class S, class T> ostream& operator << (ostream& os, const pair<S, T> v){os << "(" << v.first << ", " << v.second << ")"; return os;} template<typename T> ostream& operator<<(ostream &os, const vector<T> &v) { for (auto &e : v) os << e << ' '; return os; } template<class T> ostream& operator<<(ostream& os, const vector<vector<T>> &v){ for(auto &e : v){os << e << "\n";} return os;} struct fast_ios { fast_ios(){ cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(20); }; } fast_ios_; template <class T> void UNIQUE(vector<T> &x) {sort(ALL(x));x.erase(unique(ALL(x)), x.end());} template<class T> bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; } template<class T> bool chmin(T &a, const T &b) { if (a>b) { a=b; return 1; } return 0; } void fail() { cout << -1 << '\n'; exit(0); } inline int popcount(const int x) { return __builtin_popcount(x); } inline int popcount(const ll x) { return __builtin_popcountll(x); } template<typename T> void debug(vector<vector<T>>&v,ll h,ll w){for(ll i=0;i<h;i++) {cerr<<v[i][0];for(ll j=1;j<w;j++)cerr spa v[i][j];cerr<<"\n";}}; template<typename T> void debug(vector<T>&v,ll n){if(n!=0)cerr<<v[0]; for(ll i=1;i<n;i++)cerr spa v[i]; cerr<<"\n";}; const ll INF = (1ll<<62); // const ld EPS = 1e-10; // const ld PI = acos(-1.0); const ll mod = (int)1e9 + 7; //const ll mod = 998244353; int main(){ ll N; cin >> N; V<ll> a(N), b(N); REP(i, N) cin >> a[i]; REP(i, N) cin >> b[i]; V<V<ll>> G(N); REP(i, N-1){ ll u, v; cin >> u >> v; u--, v--; G[u].push_back(v); G[v].push_back(u); } auto dfs = [&](auto self, ll now, ll par)->pll{ V<ll> x, y, z; for(auto next: G[now]){ if(next == par) continue; pll val = self(self, next, now); x.push_back(val.first); y.push_back(val.second); z.push_back(b[now]+b[next]); } ll toru = -a[now]; ll sz = x.size(); REP(i, sz){ toru += max(x[i]+z[i], y[i]); } ll toranai = 0; REP(i, sz){ toranai += max(x[i], y[i]); } return {toru, toranai}; }; ll res = 0; REP(i, N) res += a[i]; pll val = dfs(dfs, 0, -1); res += max(val.first, val.second); cout << res << endl; return 0; }