結果

問題 No.1221 木 *= 3
ユーザー f_stshf_stsh
提出日時 2020-09-06 12:24:20
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 1,652 bytes
コンパイル時間 2,418 ms
コンパイル使用メモリ 187,272 KB
実行使用メモリ 818,176 KB
最終ジャッジ日時 2024-11-29 07:01:25
合計ジャッジ時間 22,233 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 MLE -
testcase_08 MLE -
testcase_09 MLE -
testcase_10 MLE -
testcase_11 MLE -
testcase_12 MLE -
testcase_13 MLE -
testcase_14 MLE -
testcase_15 MLE -
testcase_16 MLE -
testcase_17 MLE -
testcase_18 MLE -
testcase_19 MLE -
testcase_20 MLE -
testcase_21 MLE -
権限があれば一括ダウンロードができます

ソースコード

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(2, vector<ll>(MAX, -LINF));
vector<bool> used;
vector<int> root;
/* function */
void init() {
    cin >> N;
    a.resize(N);
    b.resize(N);
    G.resize(N, vector<int>(N));
    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