結果

問題 No.1976 Cut then Connect
ユーザー suzuken_wsuzuken_w
提出日時 2022-06-11 19:52:49
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 87 ms / 2,000 ms
コード長 2,673 bytes
コンパイル時間 3,019 ms
コンパイル使用メモリ 230,012 KB
実行使用メモリ 10,368 KB
最終ジャッジ日時 2024-05-03 12:22:15
合計ジャッジ時間 4,916 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 83 ms
9,856 KB
testcase_03 AC 41 ms
7,040 KB
testcase_04 AC 16 ms
5,376 KB
testcase_05 AC 38 ms
6,528 KB
testcase_06 AC 60 ms
8,448 KB
testcase_07 AC 41 ms
6,912 KB
testcase_08 AC 86 ms
10,240 KB
testcase_09 AC 68 ms
9,344 KB
testcase_10 AC 18 ms
5,376 KB
testcase_11 AC 87 ms
10,240 KB
testcase_12 AC 4 ms
5,376 KB
testcase_13 AC 63 ms
8,576 KB
testcase_14 AC 54 ms
8,192 KB
testcase_15 AC 51 ms
7,936 KB
testcase_16 AC 73 ms
9,728 KB
testcase_17 AC 26 ms
5,760 KB
testcase_18 AC 6 ms
5,376 KB
testcase_19 AC 57 ms
8,320 KB
testcase_20 AC 87 ms
10,368 KB
testcase_21 AC 44 ms
7,424 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 1 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 2 ms
5,376 KB
testcase_30 AC 2 ms
5,376 KB
testcase_31 AC 1 ms
5,376 KB
testcase_32 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O3")
#include<bits/stdc++.h> 
using namespace std;
using ll=long long;
using P=pair<ll,ll>;
template<class T> using V=vector<T>; 
#define fi first
#define se second
#define all(v) (v).begin(),(v).end()
const ll inf=(1e18);
// const ll mod=998244353;
const ll mod=1000000007;
const vector<int> dy={-1,0,1,0},dx={0,-1,0,1};
struct __INIT{__INIT(){cin.tie(0);ios::sync_with_stdio(false);cout<<fixed<<setprecision(15);}} __init;
template<class T> bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T> bool chmin(T &a, const T &b) { if (a>b) { a=b; return 1; } return 0; }
template<class T>void debag(const vector<T> &a){cerr<<"debag :";for(auto v:a)cerr<<v<<" ";cerr<<"\n";}
template<class T>void print(const vector<T> &a){for(auto v:a)cout<<v<<" ";cout<<"\n";}
int main(){
    int n;
    cin>>n;
    V<V<int>> g(n);
    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);
    }
    V<P> res(n);//最長,部分木の直径
    auto dfs1=[&](auto &&self,int cur,int par)->P{
        P val={0,0};
        V<ll> cp;
        for(int v:g[cur]){
            if(v==par)continue;
            P tmp=self(self,v,cur);
            cp.emplace_back(tmp.fi);
            chmax(val.fi,tmp.fi+1);
            chmax(val.se,tmp.se);
        }
        chmax(val.se,val.fi);
        if((int)cp.size()>=2){
            sort(all(cp),greater<ll>());
            chmax(val.se,cp[0]+cp[1]+2);
        }
        return res[cur]=val;
    };
    dfs1(dfs1,0,-1);
    ll ans=res[0].se;
    auto dfs2=[&](auto &&self,int cur,int par,P val)->void{
        if(par!=-1){
            chmin(ans,max({(res[cur].se+1)/2+(val.se+1)/2+1,val.se,res[cur].se}));
        }
        multiset<ll> tmp,tmp2;
        for(int v:g[cur]){
            if(v==par)continue;
            tmp.emplace(res[v].fi);
            tmp2.emplace(res[v].se);
        }
        tmp.emplace(val.fi);
        tmp2.emplace(val.se);
        for(int v:g[cur]){
            if(v==par)continue;
            P nval={0,0};
            tmp.erase(tmp.find(res[v].fi));
            tmp2.erase(tmp2.find(res[v].se));
            if((int)tmp.size()>=2){
                auto ite=tmp.end();
                ite--;ll v=*ite;
                ite--;v+=*ite;
                v+=2;
                chmax(nval.se,v);
            }
            nval.fi=*tmp.rbegin()+1;
            chmax(nval.se,*tmp2.rbegin());
            chmax(nval.se,nval.fi);
            self(self,v,cur,nval);
            tmp.emplace(res[v].fi);
            tmp2.emplace(res[v].se);
        }
    };
    dfs2(dfs2,0,-1,{-1,0});
    cout<<ans<<"\n";
}
0