結果
問題 | No.1221 木 *= 3 |
ユーザー |
![]() |
提出日時 | 2020-09-05 23:23:32 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 75 ms / 2,000 ms |
コード長 | 1,695 bytes |
コンパイル時間 | 927 ms |
コンパイル使用メモリ | 101,960 KB |
実行使用メモリ | 12,580 KB |
最終ジャッジ日時 | 2024-11-29 06:19:40 |
合計ジャッジ時間 | 3,125 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
/* -*- coding: utf-8 -*-** 1221.cc: No.1221 木 *= 3 - yukicoder*/#include<cstdio>#include<cstdlib>#include<cstring>#include<cmath>#include<iostream>#include<string>#include<vector>#include<map>#include<set>#include<stack>#include<list>#include<queue>#include<deque>#include<algorithm>#include<numeric>#include<utility>#include<complex>#include<functional>using namespace std;/* constant */const int MAX_N = 100000;const long long LINF = 1LL << 62;/* typedef */typedef long long ll;typedef vector<int> vi;typedef queue<int> qi;/* global variables */int as[MAX_N], bs[MAX_N];vi nbrs[MAX_N];int ps[MAX_N], cns[MAX_N];ll dp[MAX_N][2];/* subroutines *//* main */int main() {int n;scanf("%d", &n);for (int i = 0; i < n; i++) scanf("%d", as + i);for (int i = 0; i < n; i++) scanf("%d", bs + i);for (int i = 1; i < n; i++) {int u, v;scanf("%d%d", &u, &v);u--, v--;nbrs[u].push_back(v);nbrs[v].push_back(u);}ps[0] = -1;qi q;q.push(0);while (! q.empty()) {int u = q.front(); q.pop();int up = ps[u];vi &nbru = nbrs[u];for (vi::iterator vit = nbru.begin(); vit != nbru.end(); vit++) {int v = *vit;if (v != up) {ps[v] = u;cns[u]++;q.push(v);}}}for (int u = 0; u < n; u++)if (cns[u] == 0) q.push(u);while (! q.empty()) {int u = q.front(); q.pop();dp[u][0] += as[u];int up = ps[u];if (up >= 0) {dp[up][0] += max(dp[u][0], dp[u][1]);dp[up][1] += max(dp[u][0], dp[u][1] + bs[up] + bs[u]);if (--cns[up] == 0) q.push(up);}}printf("%lld\n", max(dp[0][0], dp[0][1]));return 0;}