結果

問題 No.1928 Make a Binary Tree
コンテスト
ユーザー t98slider
提出日時 2022-05-07 07:36:11
言語 C++14
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 213 ms / 3,000 ms
コード長 1,156 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,631 ms
コンパイル使用メモリ 177,236 KB
実行使用メモリ 52,472 KB
最終ジャッジ日時 2024-07-06 11:24:20
合計ジャッジ時間 9,285 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 57
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
using namespace std;

int main(){
    int n;
    cin >> n;
    vector<int> p(n);
    vector<vector<int>> g(n);
    vector<priority_queue<int>> t(n);
    function<int(int)> leader=[&](int v){
        if(p[v] <= 0)return v;
        return p[v] = leader(p[v]);
    };
    auto merge=[&](int u,int v){
        int x = leader(u), y = leader(v);
        if(-p[x] < -p[y]) swap(x, y);
        while(!t[y].empty()){
            t[x].push(t[y].top());
            t[y].pop();
        }
        p[x] += p[y];
        p[y] = x;
    };
    int u, v;
    for(int i = 1; i < n; i++){
        cin >> u >> v;
        g[--u].push_back(--v);
        g[v].push_back(u);
    }
    function<void(int,int)> dfs=[&](int v,int q){
        for(auto &&u:g[v]){
            if(u == q)continue;
            dfs(u, v);
            merge(u, v);
        }
        int add = 1, x = leader(v);
        for(int i = 0; i < 2; i++){
            if(!t[x].empty()){
                add += t[x].top();
                t[x].pop();
            }
        }
        t[x].push(add);
        p[x] = -t[x].size();
    };
    dfs(0, -1);
    cout << t[leader(0)].top() << endl;
}
0