結果
問題 | No.1098 LCAs |
ユーザー |
|
提出日時 | 2020-07-03 14:34:09 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 311 ms / 2,000 ms |
コード長 | 754 bytes |
コンパイル時間 | 3,184 ms |
コンパイル使用メモリ | 198,540 KB |
最終ジャッジ日時 | 2025-01-11 14:04:40 |
ジャッジサーバーID (参考情報) |
judge2 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
#define _USE_MATH_DEFINES #include <bits/stdc++.h> using namespace std; vector<vector<int>> g; vector<long long> ans; int dfs (int cur, int par) { vector<int> v; for (int child : g[cur]) if (child != par) { v.push_back(dfs(child, cur)); } int tot = accumulate(v.begin(), v.end(), 0); for (auto& i : v) ans[cur] += 1LL * i * (tot - i); ans[cur] += 2LL * tot; ans[cur]++; return tot + 1; } signed main() { ios::sync_with_stdio(false); cin.tie(0); int n; cin >> n; g = vector<vector<int>>(n); ans = vector<long long>(n); 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); } dfs(0, -1); for (auto& i : ans) cout << i << '\n'; return 0; }