結果
問題 |
No.1928 Make a Binary Tree
|
ユーザー |
|
提出日時 | 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 |
ソースコード
#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; }