結果
問題 | No.1098 LCAs |
ユーザー |
|
提出日時 | 2024-11-30 16:09:05 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 188 ms / 2,000 ms |
コード長 | 1,563 bytes |
コンパイル時間 | 3,204 ms |
コンパイル使用メモリ | 251,616 KB |
実行使用メモリ | 33,920 KB |
最終ジャッジ日時 | 2024-11-30 16:09:13 |
合計ジャッジ時間 | 7,160 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
#ifndef LOCAL#include <bits/stdc++.h>using namespace std;#define debug(...) (void(0))#else#include "algo/debug.h"#endifnamespace std {template <class Fun>class y_combinator_result {Fun fun_;public:template <class T>explicit y_combinator_result(T &&fun) : fun_(std::forward<T>(fun)) {}template <class... Args>decltype(auto) operator()(Args &&...args) {return fun_(std::ref(*this), std::forward<Args>(args)...);}};template <class Fun>decltype(auto) y_combinator(Fun &&fun) {return y_combinator_result<std::decay_t<Fun>>(std::forward<Fun>(fun));}} // namespace stdvoid solve() {int N; cin >> N;vector<vector<int>> G(N);for(int i = 1; i < N; i++) {int a, b; cin >> a >> b;a--; b--;G[a].push_back(b);G[b].push_back(a);}vector<int64_t> ans(N);vector<int> sz(N);y_combinator([&](auto self, int v, int p = -1) -> void {sz[v] = 1;for(const auto&to: G[v]) if(to != p) {self(to, v);sz[v] += sz[to];ans[v] += sz[to];}int64_t sum = sz[v] - 1;for(const auto&to: G[v]) if(to != p) {sum -= sz[to];ans[v] += sz[to] * sum;}})(0);debug(ans);for(int i = 0; i < N; i++) {ans[i] *= 2, ans[i] += 1;cout << ans[i] << '\n';}}int main() {std::ios::sync_with_stdio(false);std::cin.tie(nullptr);int tt = 1;// std::cin >> tt;while(tt--) {solve();}}