結果

問題 No.1221 木 *= 3
ユーザー yuui
提出日時 2020-09-04 22:50:51
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 209 ms / 2,000 ms
コード長 1,273 bytes
コンパイル時間 1,992 ms
コンパイル使用メモリ 197,788 KB
最終ジャッジ日時 2025-01-14 06:15:58
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); ++i)
using namespace std;
using ll = long long;
const ll INF = ll(1e18) + 5;
ll dp[int(1e5) + 5][2];
ll a[int(1e5) + 5], b[int(1e5) + 5];
vector<vector<int>> tree(int(1e5) + 5);

void dfs(int t, int p) {
    for (int child : tree[t]) {
        if (child == p) {
            continue;
        }
        dfs(child, t);
    }
    //消された場合
    dp[t][0] = a[t];
    //消されずに残った場合
    dp[t][1] = 0;
    for (int child : tree[t]) {
        if (child == p) {
            continue;
        }
        if (dp[child][0] > dp[child][1]) {
            dp[t][0] += dp[child][0];
        } else {
            dp[t][0] += dp[child][1];
        }
        if (dp[child][0] > dp[child][1] + b[t] + b[child]) {
            dp[t][1] += dp[child][0];
        } else {
            dp[t][1] += dp[child][1] + b[t] + b[child];
        }
    }
}

int main() {
    int N;
    cin >> N;
    rep(i, N) {
        cin >> a[i];
    }
    rep(i, N) {
        cin >> b[i];
    }
    rep(i, N - 1) {
        int u, v;
        cin >> u >> v;
        --u;
        --v;
        tree[u].push_back(v);
        tree[v].push_back(u);
    }
    dfs(0, -1);
    cout << max(dp[0][0], dp[0][1]) << endl;

    return 0;
}
0