# | Submission time | Handle | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
839279 | 2023-08-29T13:59:27 Z | epicci23 | Connecting Supertrees (IOI20_supertrees) | C++17 | 0 ms | 0 KB |
#include "supertrees.h" #include "bits/stdc++.h" using namespace std; #define pb push_back #define endl "\n" #define int long long #define sz(x) ((int)(x).size()) #define all(x) (x).begin(),(x).end() const int N = 1005; int par[N]; int siz[N]; int find(int a) { if(par[a]==a) return a; return par[a]=find(par[a]); } void unite(int a,int b) { a=find(a),b=find(b); if(a==b) return; if(siz[a]>siz[b]) swap(a,b); siz[b]+=siz[a]; par[a]=b; } void build(vector<vector<int>> g) { int lol=sz(g); for(int i=0;i<lol;i++) for(int j=0;j<lol;j++) cout << g[i][j] << " \n"[j==lol-1]; } int construct(vector<vector<int>> p) { iota(par,par+N,0LL); fill(siz,siz+N,1LL); int n = p[0].size(); vector<vector<int>> g(n,vector<int>(n,0)); for(int i=0;i<n;i++) for(int j=0;j<n;j++) if(p[i][j]) unite(i,j); for(int i=0;i<n;i++) for(int j=0;j<n;j++) if(!p[i][j] && find(i)==find(j)) return 0; vector<int> no[n]; for(int i=0;i<n;i++) no[find(i)].pb(i); for(int xd=0;xd<n;xd++) { if(no[xd].size()<=1) continue; int m = no[xd].size(); vector<int> v[m]; for(int i=0;i<m;i++) { for(int j=i;j<m;j++) { if(i==j) { v[i].pb(i); continue; } int a=no[xd][i]; int b=no[xd][j]; if(p[a][b]==1) { v[i].pb(j); v[j].pb(i); } } } for(int i=0;i<m;i++) sort(all(v[i])); vector<int> tems; vector<int> vis(m,0LL); for(int i=0;i<m;i++) { if(vis[i]) continue; tems.pb(i); for(int x:v[i]) { if(vis[x]) return 0; vis[x]=1; if(v[i]!=v[x]) return 0; } for(int j=1;j<sz(v[i]);j++) { int a=no[xd][v[i][j]]; int b=no[xd][v[i][j-1]]; g[a][b]=g[b][a]=1; } } int k=sz(tems); if(k==2) return 0; if(k>2) { for(int i=1;i<k;i++) { g[no[xd][tems[i]]][no[xd][tems[i-1]]]=1; g[no[xd][tems[i-1]]][no[xd][tems[i]]]=1; } g[no[xd][tems[k-1]]][no[xd][tems[0]]]=1; g[no[xd][tems[0]]][no[xd][tems[k-1]]]=1; } } build(g); return 1; }