結果

問題 No.1928 Make a Binary Tree
ユーザー Nanashima
提出日時 2022-05-06 23:45:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 185 ms / 3,000 ms
コード長 1,226 bytes
コンパイル時間 1,832 ms
コンパイル使用メモリ 177,980 KB
実行使用メモリ 43,084 KB
最終ジャッジ日時 2024-07-06 03:18:53
合計ジャッジ時間 8,681 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 57
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#define rep(i,n) for(int i=0; i<(n); i++)
#define INF ((1LL<<62)-(1LL<<31))
#define all(a)  (a).begin(),(a).end()
#define rall(a)  (a).rbegin(),(a).rend()
typedef long long ll;
typedef pair<ll,ll> pl;

ll ans=1;

priority_queue<ll,vector<ll>> dfs(vector<vector<int>> &g,int pos,int par) {
    ll cnt=0,ret=1;
    priority_queue<ll,vector<ll>> pq;
    for(auto nv:g[pos]) {
        if(nv==par) continue;
        cnt++;
        priority_queue<ll,vector<ll>> pq2=dfs(g,nv,pos);
        if((int)pq2.size()>(int)pq.size()) swap(pq,pq2);
        while(!pq2.empty()) {
            pq.push(pq2.top());
            pq2.pop();
        }
    }
    if(cnt==0) {
        pq.push(1);
        return pq;
    }
    rep(i,2) {
        if(pq.empty()) break;
        ret+=pq.top();
        pq.pop();
    }
    if(pos==0) ans=ret;
    pq.push(ret);
    return pq;
}

int main() {
    ll n;
    cin >> n;
    vector<vector<int>> g(n,vector<int> ());
    vector<int> deg(n,0);
    rep(i,n-1) {
        int u,v;
        cin >> u >> v;
        u--; v--;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    priority_queue<ll,vector<ll>> pq=dfs(g,0,-1);
    cout << ans << endl;
    return 0;
}
0