結果
問題 |
No.3206 う し た ウ ニ 木 あ く ん 笑
|
ユーザー |
![]() |
提出日時 | 2025-07-17 13:32:59 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 537 ms / 3,000 ms |
コード長 | 6,090 bytes |
コンパイル時間 | 6,672 ms |
コンパイル使用メモリ | 335,800 KB |
実行使用メモリ | 179,016 KB |
最終ジャッジ日時 | 2025-07-17 13:33:16 |
合計ジャッジ時間 | 13,454 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
#include <bits/stdc++.h> using namespace std; using std::cerr; using std::cin; using std::cout; #if __has_include(<atcoder/all>) #include <atcoder/all> using mint = atcoder::modint998244353; istream &operator>>(istream &is, mint &a) { long long t; is >> t; a = t; return is; } ostream &operator<<(ostream &os, mint a) { return os << a.val(); } #endif typedef long double ld; #define long long long #define uint unsigned int #define ull unsigned long #define overload3(a, b, c, name, ...) name #define rep3(i, a, b) for (int i = (a); i < (b); i++) #define rep2(i, n) rep3(i, 0, n) #define rep1(n) rep2(__i, n) #define rep(...) overload3(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__) #define per3(i, a, b) for (int i = (b) - 1; i >= (a); i--) #define per2(i, n) per3(i, 0, n) #define per1(n) per2(__i, n) #define per(...) overload3(__VA_ARGS__, per3, per2, per1)(__VA_ARGS__) #define all(a) a.begin(), a.end() #define UNIQUE(a) \ sort(all(a)); \ a.erase(unique(all(a)), a.end()) #define sz(a) (int)a.size() #define vec vector #ifndef DEBUG #define cerr \ if (0) \ cerr // #undef assert // #define assert(...) void(0) #undef endl #define endl '\n' #endif template <typename T> ostream &operator<<(ostream &os, vector<T> a) { const int n = a.size(); rep(i, n) { os << a[i]; if (i + 1 != n) os << " "; } return os; } template <typename T, size_t n> ostream &operator<<(ostream &os, array<T, n> a) { rep(i, n) { os << a[i]; if (i + 1 != n) os << " "; } return os; } template <typename T> istream &operator>>(istream &is, vector<T> &a) { for (T &i : a) is >> i; return is; } template <typename T, typename S> bool chmin(T &x, S y) { if ((T)y < x) { x = (T)y; return true; } return false; } template <typename T, typename S> bool chmax(T &x, S y) { if (x < (T)y) { x = (T)y; return true; } return false; } template <typename T> void operator++(vector<T> &a) { for (T &i : a) ++i; } template <typename T> void operator--(vector<T> &a) { for (T &i : a) --i; } template <typename T> void operator++(vector<T> &a, int) { for (T &i : a) i++; } template <typename T> void operator--(vector<T> &a, int) { for (T &i : a) i--; } /* r を根として固定 V dfs(vertex v){ E prod = e(); for ( c in child(v) ){ prod = merge(prod, put_edge(dfs(c),idx(v,c))); } return put_vertex(prod, v); } r=0,...,n-1 に対し r を根として dfs(r) の値を求める */ // add_edge -> build -> reroot に答え template <class E, class V, E (*merge)(E, E), E (*e)(), E (*put_edge)(V, int), V (*put_vertex)(E, int)> struct RerootingDP { struct edge { int to, idx, xdi; }; RerootingDP(int n_ = 0) : n(n_), inner_edge_id(0) { es.resize(2 * n - 2); start.resize(2 * n - 2); if (n == 1) es_build(); } void add_edge(int u, int v, int idx, int xdi) { start[inner_edge_id] = u; es[inner_edge_id] = {v, idx, xdi}; inner_edge_id++; start[inner_edge_id] = v; es[inner_edge_id] = {u, xdi, idx}; inner_edge_id++; if (inner_edge_id == 2 * n - 2) { es_build(); } } vector<V> build(int root_ = 0) { root = root_; vector<V> subdp(n); subdp[0] = put_vertex(e(), 0); outs.resize(n); vector<int> geta(n + 1, 0); for (int i = 0; i < n; i++) geta[i + 1] = start[i + 1] - start[i] - 1; geta[root + 1]++; for (int i = 0; i < n; i++) geta[i + 1] += geta[i]; auto dfs = [&](auto sfs, int v, int f) -> void { E val = e(); for (int i = start[v]; i < start[v + 1]; i++) { if (es[i].to == f) { swap(es[start[v + 1] - 1], es[i]); } if (es[i].to == f) continue; sfs(sfs, es[i].to, v); E nval = put_edge(subdp[es[i].to], es[i].idx); outs[geta[v]++] = nval; val = merge(val, nval); } subdp[v] = put_vertex(val, v); }; dfs(dfs, root, -1); return subdp; } vector<V> reroot() { vector<E> reverse_edge(n); reverse_edge[root] = e(); vector<V> answers(n); auto dfs = [&](auto sfs, int v) -> void { int le = outs_start(v); int ri = outs_start(v + 1); int siz = ri - le; vector<E> rui(siz + 1); rui[siz] = e(); for (int i = siz - 1; i >= 0; i--) { rui[i] = merge(outs[le + i], rui[i + 1]); } answers[v] = put_vertex(merge(rui[0], reverse_edge[v]), v); E lui = e(); for (int i = 0; i < siz; i++) { V rdp = put_vertex( merge(merge(lui, rui[i + 1]), reverse_edge[v]), v); reverse_edge[es[start[v] + i].to] = put_edge(rdp, es[start[v] + i].xdi); lui = merge(lui, outs[le + i]); sfs(sfs, es[start[v] + i].to); } }; dfs(dfs, root); return answers; } private: int n, root, inner_edge_id; vector<E> outs; vector<edge> es; vector<int> start; int outs_start(int v) { int res = start[v] - v; if (root < v) res++; return res; } void es_build() { vector<edge> nes(2 * n - 2); vector<int> nstart(n + 2, 0); for (int i = 0; i < 2 * n - 2; i++) nstart[start[i] + 2]++; for (int i = 0; i < n; i++) nstart[i + 1] += nstart[i]; for (int i = 0; i < 2 * n - 2; i++) nes[nstart[start[i] + 1]++] = es[i]; swap(es, nes); swap(start, nstart); } }; using vi = vec<int>; vi merge(vi a, vi b) { for (auto i : b) a.push_back(i); return a; } vi e() { return {0}; } vi put_edge(vi a, int) { return {1 + *max_element(all(a))}; } vi put_vertex(vi a, int) { return a; } void solve() { int n; cin >> n; RerootingDP<vi, vi, merge, e, put_edge, put_vertex> g(n); rep(i, n - 1) { int u, v; cin >> u >> v; u--, v--; g.add_edge(u, v, -1, -1); } g.build(); auto res = g.reroot(); long ans = 0; for (vi &a : res) { sort(all(a), greater<int>()); rep(i, sz(a)) { chmax(ans, (i + 1) * a[i]); } } cout << ans + 1 << endl; } int main() { // srand((unsigned)time(NULL)); cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(20); int t = 1; // cin >> t; while (t--) solve(); }