結果

問題 No.1221 木 *= 3
ユーザー tonyu0
提出日時 2020-09-04 22:40:39
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 880 bytes
コンパイル時間 1,373 ms
コンパイル使用メモリ 101,900 KB
最終ジャッジ日時 2025-01-14 06:05:37
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 8 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cmath>
#include <iostream>
#include <map>
#include <numeric>
#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<int> 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<bool> deleted(n);
  rep(i, 0, n) {
    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