結果
問題 | No.1098 LCAs |
ユーザー | 👑 emthrm |
提出日時 | 2020-06-26 21:49:44 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 200 ms / 2,000 ms |
コード長 | 1,441 bytes |
コンパイル時間 | 1,950 ms |
コンパイル使用メモリ | 198,980 KB |
最終ジャッジ日時 | 2025-01-11 11:15:23 |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
#define _USE_MATH_DEFINES #include <bits/stdc++.h> using namespace std; #define FOR(i,m,n) for(int i=(m);i<(n);++i) #define REP(i,n) FOR(i,0,n) #define ALL(v) (v).begin(),(v).end() using ll = long long; const int INF = 0x3f3f3f3f; const ll LINF = 0x3f3f3f3f3f3f3f3fLL; const double EPS = 1e-8; const int MOD = 1000000007; // const int MOD = 998244353; const int dy[] = {1, 0, -1, 0}, dx[] = {0, -1, 0, 1}; const int dy8[] = {1, 1, 0, -1, -1, -1, 0, 1}, dx8[] = {0, -1, -1, -1, 0, 1, 1, 1}; template <typename T, typename U> inline bool chmax(T &a, U b) { return a < b ? (a = b, true) : false; } template <typename T, typename U> inline bool chmin(T &a, U b) { return a > b ? (a = b, true) : false; } struct IOSetup { IOSetup() { cin.tie(nullptr); ios_base::sync_with_stdio(false); cout << fixed << setprecision(20); } } iosetup; int main() { int n; cin >> n; vector<vector<int>> graph(n); REP(_, n - 1) { int v, w; cin >> v >> w; --v; --w; graph[v].emplace_back(w); graph[w].emplace_back(v); } vector<int> par(n, -1), sub(n, 1); function<void(int)> dfs = [&](int ver) { for (int e : graph[ver]) { if (e != par[ver]) { par[e] = ver; dfs(e); sub[ver] += sub[e]; } } }; dfs(0); REP(i, n) { ll ans = sub[i]; for (int e : graph[i]) { if (e != par[i]) ans += 1LL * sub[e] * (sub[i] - sub[e]); } cout << ans << '\n'; } return 0; }