結果

問題 No.1221 木 *= 3
ユーザー tonyu0tonyu0
提出日時 2020-09-04 22:54:17
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,157 bytes
コンパイル時間 1,409 ms
コンパイル使用メモリ 121,960 KB
実行使用メモリ 13,672 KB
最終ジャッジ日時 2024-11-26 20:49:47
合計ジャッジ時間 4,986 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <algorithm>
#include <cmath>
#include <iomanip>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <tuple>
#include <vector>
using namespace std;
using ll = int64_t;
#define rep(i, j, n) for (int i = j; i < (int)n; ++i)
#define rrep(i, j, n) for (int i = (int)n - 1; j <= i; --i)

int main() {
  int n;
  cin >> n;
  vector<ll> a(n), b(n);
  rep(i, 0, n) cin >> a[i];
  rep(i, 0, n) cin >> b[i];
  vector<vector<int>> g(n);
  vector<ll> con(n);
  rep(i, 0, n - 1) {
    int u, v;
    cin >> u >> v;
    --u, --v;
    g[u].push_back(v);
    g[v].push_back(u);
    con[u]++;
    con[v]++;
  }

  ll all = 0;
  rep(i, 0, n) all += con[i] * b[i];

  vector<pair<ll, int>> gain;
  rep(i, 0, n) {
    ll add = a[i];
    for (int j : g[i]) add -= b[i] + b[j];
    gain.emplace_back(add, i);
  }
  sort(gain.rbegin(), gain.rend());

  vector<bool> deleted(n);
  for (auto e : gain) {
    int i = e.second;
    ll add = a[i];
    for (int j : g[i]) {
      if (!deleted[j]) { add -= b[i] + b[j]; }
    }
    if (add > 0) {
      all += add;
      deleted[i] = true;
    }
  }
  cout << all << '\n';

  return 0;
}
0