結果

問題 No.3206 う　し　た　ウ　ニ　木　あ　く　ん　笑
ユーザー pitP
提出日時 2025-07-18 22:26:21
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 538 ms / 3,000 ms
コード長 5,524 bytes
コンパイル時間 6,363 ms
コンパイル使用メモリ 334,252 KB
実行使用メモリ 196,328 KB
最終ジャッジ日時 2025-07-18 22:26:37
合計ジャッジ時間 13,864 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
istream &operator>>(istream &is, modint &a) { long long v; is >> v; a = v; return is; }
ostream &operator<<(ostream &os, const modint &a) { return os << a.val(); }
istream &operator>>(istream &is, modint998244353 &a) { long long v; is >> v; a = v; return is; }
ostream &operator<<(ostream &os, const modint998244353 &a) { return os << a.val(); }
istream &operator>>(istream &is, modint1000000007 &a) { long long v; is >> v; a = v; return is; }
ostream &operator<<(ostream &os, const modint1000000007 &a) { return os << a.val(); } 

typedef long long ll;
typedef vector<vector<int>> Graph;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
#define FOR(i,l,r) for (int i = l;i < (int)(r); i++)
#define rep(i,n) for (int i = 0;i < (int)(n); i++)
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define my_sort(x) sort(x.begin(), x.end())
#define my_max(x) *max_element(all(x))
#define my_min(x) *min_element(all(x))
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; }
const int INF = (1<<30) - 1;
const ll LINF = (1LL<<62) - 1;
const int MOD = 998244353;
const int MOD2 = 1e9+7;
const double PI = acos(-1);
vector<int> di = {1,0,-1,0};
vector<int> dj = {0,1,0,-1};

#ifdef LOCAL
#  include <debug_print.hpp>
#  define debug(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__)
#else
#  define debug(...) (static_cast<void>(0))
#endif

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 from, to, idx, rev_idx;
    };
    RerootingDP (int _n = 0) : n(_n) { es.resize(n);}
    void add_edge(int u, int v, int idx1, int idx2){
        es[u].push_back({u,v,idx1,idx2});
        es[v].push_back({v,u,idx2,idx1});
    }
    V build(int v = 0){
        root = v;
        vis.resize(n,0);
        outs.resize(n);
        return dfs(root);
    }
    vector<V> reroot(){
        reverse_edge.resize(n);
        reverse_edge[root] = e();
        answers.resize(n);
        bfs(root);
        return answers;
    }
  private:
    int n, root;
    vector<vector<edge>> es;
    vector<int> vis;
    vector<vector<E>> outs;
    vector<E> reverse_edge;
    vector<V> answers;
    V dfs(int v){
        vis[v]++;
        E val = e();
        for (auto &p : es[v]){
            if (vis[p.to] > 0 && p.to != es[v].back().to) swap(p,es[v].back());
            if (vis[p.to] > 0) continue;
            E nval = put_edge(dfs(p.to),p.idx);
            outs[v].emplace_back(nval);
            val = merge(val,nval);
        }
        return put_vertex(val,v);
    }
    void bfs(int v){
        int siz = outs[v].size();
        vector<E> lui(siz+1), rui(siz+1);
        lui[0] = e(), rui[siz] = e();
        for (int i = 0; i < siz; i++) lui[i+1] = merge(lui[i],outs[v][i]);
        for (int i = siz-1; i >= 0; i--) rui[i] = merge(outs[v][i],rui[i+1]);
        for (int i = 0; i < siz; i++){
            reverse_edge[es[v][i].to] = put_edge(put_vertex(merge(merge(lui[i],rui[i+1]),reverse_edge[v]),v),es[v][i].rev_idx);
            bfs(es[v][i].to);
        }
        answers[v] = put_vertex(merge(lui[siz],reverse_edge[v]), v);
    }
};

// Education dp subtree
// https://atcoder.jp/contests/dp/submissions/37662999
using T_monoid = pair<vector<int>, int>; // E
using T_DP = pair<int, int>; // V

T_monoid merge(T_monoid a, T_monoid b){
    if((int)a.first.size() < (int)b.first.size()){
        for(auto &v : b.first){
            a.first.push_back(v);
        }
        return a;
    }
    else{
        for(auto &v : a.first){
            b.first.push_back(v);
        }
        return b;
    }
}
T_monoid e(){
    return make_pair(vector<int>{}, -1);
}
T_monoid put_edge(T_DP x, int i){
    return make_pair(vector<int>{x.first}, x.second);
}
T_DP put_vertex(T_monoid x, int v){
    if(x.first.empty()){
        return make_pair(1, 0);
    }
    pii ret = {0, 0};
    sort(all(x.first));
    rep(i, (int)x.first.size()) chmax(ret.second, x.first[i] * ((int)x.first.size() - i));
    ret.first = x.first.back() + 1;
    return ret;
}
// 参考
// https://trap.jp/post/1702/

// https://null-mn.hatenablog.com/entry/2020/04/14/124151 
// fがput_edge, gがput_vertex
// 辺の情報が必要なときはedge[i]を作る、また頂点の情報が必要な時はvertex[v]を作る


int main(){
    cin.tie(0);
    ios_base::sync_with_stdio(false);

    int N; cin >> N;
    Graph g(N);
    RerootingDP<T_monoid, T_DP, merge, e, put_edge, put_vertex> dp(N);

    rep(i, N - 1){
        int u, v; 
        cin >> u >> v;
        u--, v--;

        g[u].push_back(v);
        g[v].push_back(u);

        dp.add_edge(u, v, i, i);
    }

    // {dist, ans}
    function<pii(int, int)> dfs = [&](int now, int par) {
        vector<int> ds;
        for(int nxt : g[now]){
            if(nxt == par) continue;

            pii chr = dfs(nxt, now);
            ds.push_back(chr.first);
        }
        sort(all(ds));
        if(ds.empty()){
            return make_pair(1, 0);
        }

        pii ret = {ds.back() + 1, 0};
        rep(i, (int)ds.size()) chmax(ret.second, ds[i] * ((int)ds.size() - i));

        return ret;
    };

    dp.build();
    vector<T_DP> ret = dp.reroot();
    debug(ret);
    int ans = 0;
    rep(i, N) chmax(ans, ret[i].second);
    cout << ans + 1 << endl;
}
0