Submission #911141

#	Time	Username	Problem	Language	Result	Execution time	Memory
911141		ttamx	Network (BOI15_net)	C++14	100 / 100	639 ms	70808 KiB

net

#include<bits/stdc++.h>

using namespace std;

const int N=5e5+5;

int n;
int sz[N];
vector<int> adj[N];

int dfssz(int u,int p=0){
    sz[u]=(adj[u].size()<2);
    for(auto v:adj[u])if(v!=p)sz[u]+=dfssz(v,u);
    return sz[u];
}

int centroid(int u,int cnt,int p=0){
    for(auto v:adj[u])if(v!=p&&sz[v]>cnt/2)return centroid(v,cnt,u);
    return u;
}

void dfs(int u,int p,vector<int> &comp){
    if(adj[u].size()==1)comp.emplace_back(u);
    for(auto v:adj[u])if(v!=p)dfs(v,u,comp);
}

int main(){
    cin.tie(nullptr)->sync_with_stdio(false);
    cin >> n;
    for(int i=1;i<n;i++){
        int u,v;
        cin >> u >> v;
        adj[u].emplace_back(v);
        adj[v].emplace_back(u);
    }
    int c=centroid(1,dfssz(1));
    vector<vector<int>> res;
    int cnt=0;
    for(auto v:adj[c]){
        res.push_back({});
        dfs(v,c,res.back());
    }
    int m=res.size();
    priority_queue<pair<int,int>> pq;
    for(int i=0;i<m;i++)pq.emplace(res[i].size(),i);
    vector<pair<int,int>> ans;
    while(!pq.empty()){
        auto [s,i]=pq.top();
        pq.pop();
        if(pq.empty()){
            ans.emplace_back(res[i].back(),c);
            break;
        }
        auto [t,j]=pq.top();
        pq.pop();
        ans.emplace_back(res[i].back(),res[j].back());
        res[i].pop_back();
        res[j].pop_back();
        if(--s)pq.emplace(s,i);
        if(--t)pq.emplace(t,j);
    }
    cout << ans.size() << "\n";
    for(auto [x,y]:ans)cout << x << " " << y << "\n";
}

Compilation message (stderr)

net.cpp: In function 'int main()':
net.cpp:48:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   48 |         auto [s,i]=pq.top();
      |              ^
net.cpp:54:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   54 |         auto [t,j]=pq.top();
      |              ^
net.cpp:63:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   63 |     for(auto [x,y]:ans)cout << x << " " << y << "\n";
      |              ^
net.cpp:38:9: warning: unused variable 'cnt' [-Wunused-variable]
   38 |     int cnt=0;
      |         ^~~

• Subtask 1: 18 / 18
• Subtask 2: 45 / 45
• Subtask 3: 37 / 37