結果

問題 No.1928 Make a Binary Tree
コンテスト
ユーザー Nanashima
提出日時 2022-05-06 23:44:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,226 bytes
コンパイル時間 1,913 ms
コンパイル使用メモリ 177,412 KB
実行使用メモリ 43,008 KB
最終ジャッジ日時 2024-07-06 03:17:02
合計ジャッジ時間 9,145 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 56 WA * 1
権限があれば一括ダウンロードができます

ソースコード

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=0;

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