結果
問題 | No.1098 LCAs |
ユーザー | siro53 |
提出日時 | 2020-06-26 23:19:40 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 287 ms / 2,000 ms |
コード長 | 2,336 bytes |
コンパイル時間 | 2,931 ms |
コンパイル使用メモリ | 208,088 KB |
実行使用メモリ | 54,100 KB |
最終ジャッジ日時 | 2023-09-18 07:58:23 |
合計ジャッジ時間 | 7,958 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,376 KB |
testcase_01 | AC | 1 ms
4,380 KB |
testcase_02 | AC | 1 ms
4,376 KB |
testcase_03 | AC | 1 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,380 KB |
testcase_05 | AC | 2 ms
4,380 KB |
testcase_06 | AC | 2 ms
4,376 KB |
testcase_07 | AC | 1 ms
4,376 KB |
testcase_08 | AC | 1 ms
4,380 KB |
testcase_09 | AC | 2 ms
4,376 KB |
testcase_10 | AC | 1 ms
4,376 KB |
testcase_11 | AC | 2 ms
4,376 KB |
testcase_12 | AC | 2 ms
4,380 KB |
testcase_13 | AC | 2 ms
4,384 KB |
testcase_14 | AC | 2 ms
4,376 KB |
testcase_15 | AC | 2 ms
4,376 KB |
testcase_16 | AC | 2 ms
4,376 KB |
testcase_17 | AC | 2 ms
4,376 KB |
testcase_18 | AC | 212 ms
16,400 KB |
testcase_19 | AC | 212 ms
16,396 KB |
testcase_20 | AC | 204 ms
16,708 KB |
testcase_21 | AC | 202 ms
16,536 KB |
testcase_22 | AC | 205 ms
16,644 KB |
testcase_23 | AC | 143 ms
19,328 KB |
testcase_24 | AC | 143 ms
19,420 KB |
testcase_25 | AC | 127 ms
19,496 KB |
testcase_26 | AC | 135 ms
19,376 KB |
testcase_27 | AC | 135 ms
19,488 KB |
testcase_28 | AC | 284 ms
50,924 KB |
testcase_29 | AC | 287 ms
54,100 KB |
testcase_30 | AC | 285 ms
50,728 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; template <class T> inline bool chmax(T &a, T b) { if(a < b) { a = b; return 1; } return 0; } template <class T> inline bool chmin(T &a, T b) { if(a > b) { a = b; return 1; } return 0; } void debug() { cerr << "\n"; } template <class T> void debug(const T &x) { cerr << x << "\n"; } template <class T, class... Args> void debug(const T &x, const Args &... args) { cerr << x << " "; debug(args...); } template <class T> void debugVector(const vector<T> &v) { for(const T &x : v) { cerr << x << " "; } cerr << "\n"; } using ll = long long; #define ALL(v) (v).begin(), (v).end() #define RALL(v) (v).rbegin(), (v).rend() const double EPS = 1e-7; const int INF = 1 << 30; const ll LLINF = 1LL << 60; const double PI = acos(-1); constexpr int MOD = 1000000007; const int dx[4] = {1, 0, -1, 0}; const int dy[4] = {0, 1, 0, -1}; //------------------------------------- int n; vector<vector<int>> g; vector<int> sz; vector<ll> ans; int dfs1(int v, int p) { if(sz[v] != -1) { return sz[v]; } int res = 1; for(int u : g[v]) { if(u != p) { res += dfs1(u, v); } } return (sz[v] = res); } void dfs2(int v, int p) { ll res = 1; vector<int> tmp; for(int u : g[v]) { if(u != p) { tmp.emplace_back(u); } } int N = tmp.size(); vector<ll> sum(N + 1, 0); for(int i = 0; i < N; i++) { sum[i + 1] = sum[i] + sz[tmp[i]]; } // debug("now:", v); // debugVector(tmp); // debugVector(mul); for(int i = 0; i < N; i++) { res += 2LL * (sum[N] - sum[i + 1]) * sz[tmp[i]]; } // debug(res - 1); for(int u : g[v]) { if(u == p) { continue; } res += 2LL * sz[u]; dfs2(u, v); } ans[v] = res; } int main() { cin.tie(0); ios::sync_with_stdio(false); cin >> n; g.resize(n), sz.resize(n, -1), ans.resize(n, 0); for(int i = 0; i < n - 1; i++) { int a, b; cin >> a >> b; a--; b--; g[a].emplace_back(b); g[b].emplace_back(a); } dfs1(0, -1); dfs2(0, -1); // debugVector(sz); for(int i = 0; i < n; i++) { cout << ans[i] << "\n"; } }