結果

問題 No.3206 う し た ウ ニ 木 あ く ん 笑
ユーザー Today03
提出日時 2025-07-18 21:46:32
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 191 ms / 3,000 ms
コード長 2,260 bytes
コンパイル時間 3,830 ms
コンパイル使用メモリ 288,272 KB
実行使用メモリ 90,612 KB
最終ジャッジ日時 2025-07-18 21:46:53
合計ジャッジ時間 6,749 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

/*
    葉までの距離の最大値をもっておいて全方位?
*/

#include <bits/stdc++.h>
using namespace std;
#define ALL(x) (x).begin(),(x).end()
#define REP(i, n) for(ll i=0; i<(ll)(n); i++)

template<typename T> int LB(const vector<T>& v, T x) { return lower_bound(ALL(v),x)-v.begin(); }
template<typename T> int UQ(T& v) { sort(ALL(v)); v.erase(unique(ALL(v)),v.end()); return v.size(); }
template<typename T> bool chmax(T &a, T b) { return a<b ? a=b, true : false; }
template<typename T> bool chmin(T &a, T b) { return a>b ? a=b, true : false; }
template<typename T> using rpriority_queue = priority_queue<T,vector<T>,greater<T>>;
using ll=long long; const int INF=1e9+10; const ll INFL=4e18;
using ld=long double; using lll=__int128_t; using ull=unsigned long long;
using VI=vector<int>; using VVI=vector<VI>; using VL=vector<ll>; using VVL=vector<VL>;
using PL=pair<ll,ll>; using VP=vector<PL>; using WG=vector<vector<pair<int,ll>>>;


#ifdef LOCAL
#include "./debug.hpp"
#else
#define debug(...)
#define print_line
#endif


//----------------------------------------------------------

void solve() {
    ll N; cin>>N;
    VVI G(N);
    REP(i,N-1) {
        ll u,v; cin>>u>>v; u--; v--;
        G[u].push_back(v); G[v].push_back(u);
    }

    VL dp(N);
    auto dfs=[&](auto&& dfs, int now, int pre)-> void {
        for(int nxt: G[now]) if(nxt!=pre) {
            dfs(dfs,nxt,now);
            chmax(dp[now],dp[nxt]);
        }
        dp[now]++;
    };
    dfs(dfs,0,-1);

    ll ans=0;
    auto dfs2=[&](auto&& dfs2, int now, int pre, ll prp)-> void {
        ll d=G[now].size();
        VL dp2(d);
        REP(i,d) dp2[i]=G[now][i]==pre?prp:dp[G[now][i]];

        auto A=dp2;
        sort(ALL(A)); reverse(ALL(A));
        REP(i,d) chmax(ans,(i+1)*A[i]);

        VL dpl(d+1,-INF), dpr(d+1,-INF);
        REP(i,d) dpl[i+1]=max(dpl[i],dp2[i]);
        for(int i=d-1; i>=0; i--) dpr[i]=max(dpr[i+1],dp2[i]);

        REP(i,d) if(G[now][i]!=pre) {
            dfs2(dfs2,G[now][i],now,max(dpl[i],dpr[i+1])+1);
        }
    };
    dfs2(dfs2,0,-1,0);
    ans++;

    cout<<ans<<endl;
}

int main() {
    ios::sync_with_stdio(false); cin.tie(nullptr);
    //cout<<fixed<<setprecision(15);
    int T=1;
    //cin>>T;
    while(T--) solve();
}
0