結果

問題 No.1221 木 *= 3
ユーザー f_stshf_stsh
提出日時 2020-09-06 12:27:22
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 178 ms / 2,000 ms
コード長 1,627 bytes
コンパイル時間 2,011 ms
コンパイル使用メモリ 186,292 KB
実行使用メモリ 19,668 KB
最終ジャッジ日時 2023-08-19 13:49:35
合計ジャッジ時間 6,217 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 AC 1 ms
4,376 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 171 ms
19,488 KB
testcase_08 AC 172 ms
19,112 KB
testcase_09 AC 150 ms
19,668 KB
testcase_10 AC 159 ms
19,160 KB
testcase_11 AC 157 ms
19,504 KB
testcase_12 AC 143 ms
12,696 KB
testcase_13 AC 135 ms
12,760 KB
testcase_14 AC 138 ms
12,772 KB
testcase_15 AC 139 ms
13,068 KB
testcase_16 AC 144 ms
12,696 KB
testcase_17 AC 156 ms
12,492 KB
testcase_18 AC 174 ms
12,436 KB
testcase_19 AC 171 ms
12,516 KB
testcase_20 AC 175 ms
12,560 KB
testcase_21 AC 178 ms
12,504 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;
vector<bool> used;
// vector<int> root;
/* 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) {
    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 (used[y]) continue;
        dfs(y);
        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));
    used.assign(N, false);

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