結果

問題 No.1221 木 *= 3
ユーザー f_stshf_stsh
提出日時 2020-09-06 12:46:15
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 178 ms / 2,000 ms
コード長 1,582 bytes
コンパイル時間 2,023 ms
コンパイル使用メモリ 183,568 KB
実行使用メモリ 19,684 KB
最終ジャッジ日時 2024-05-06 20:57:51
合計ジャッジ時間 5,751 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 174 ms
19,684 KB
testcase_08 AC 178 ms
19,556 KB
testcase_09 AC 175 ms
19,684 KB
testcase_10 AC 176 ms
19,436 KB
testcase_11 AC 178 ms
19,564 KB
testcase_12 AC 164 ms
12,948 KB
testcase_13 AC 159 ms
12,952 KB
testcase_14 AC 154 ms
13,080 KB
testcase_15 AC 156 ms
12,952 KB
testcase_16 AC 162 ms
12,948 KB
testcase_17 AC 174 ms
12,672 KB
testcase_18 AC 171 ms
12,672 KB
testcase_19 AC 176 ms
12,672 KB
testcase_20 AC 173 ms
12,800 KB
testcase_21 AC 173 ms
12,800 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
/* typedef */
typedef long long ll;
typedef pair<int, int> pii;
/* constant */
const int INF = 1 << 30;
const ll LINF = 1LL << 50;
const int NIL = -1;
const int MAX = 10000;
const int mod = 1000000007;
const double pi = 3.141592653589;
/* global variables */
int N;
vector<ll> a, b;
vector<vector<int>> G;
vector<vector<ll>> dp;
/* function */
void init() {
    cin >> N;
    a.resize(N);
    b.resize(N);
    G.resize(N, vector<int>());
    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].push_back(v);
        G[v].push_back(u);
    }
}
void dfs(int x, int par) {
    // if (used[x]) return ;
    // used[x] = true;
    // transition
    // dp[0][i] = a[i] + sum_{j = i.child} [max(dp[0][j], dp[1][j])]
    // dp[1][i] = sum_{j = i.child}[dp[0][j], dp[1][j] + b[i] + b[j]]
    dp[0][x] = a[x];
    dp[1][x] = 0;
    for (int y : G[x]) {
        if (par == y) continue;
        dfs(y, x);
        dp[0][x] += max(dp[0][y], dp[1][y]);
        dp[1][x] += max(dp[0][y], dp[1][y] + b[x] + b[y]);
    }
}
void solve() {
    // dp[0][i]: 頂点iを根とする部分木について,頂点iを消した場合のスコアの最大値
    // dp[1][i]: 頂点iを根とする部分木について,頂点iを残した場合のスコアの最大値
    dp.assign(2, vector<ll>(N, 0));

    dfs(0, -1);
    cout << max(dp[0][0], dp[1][0]) << '\n';
}
/* main */
int main(){
    init();
    solve();
}
0