# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
653771 | coding_snorlax | Connecting Supertrees (IOI20_supertrees) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "supertrees.h"
#include<bits/stdc++.h>
using namespace std;
int Boss[1005];
int Rank[1005];
int query(int x){
if(Boss[x]==x) return x;
else return Boss[x]=query(Boss[x]);
}
int Union(int x,int y){
int A1=query(x);
int A2=query(y);
if(A1==A2){
return 0;
}
if (Rank[A1]>Rank[A2]) Boss[A2]=A1;
else if(Rank[A1]<Rank[A2]) Boss[A1]=A2;
else{
Boss[A2]=A1;
Rank[A1]++;
}
return 1;
}
int construct(vector<vector<int>> x){
for(int i=0;i<1005;i++){
Boss[i]=i;
Rank[i]=0;
}
vector<vector<int>> built;
for(int i=0;i<x.size();i++){
vector<int> New;
for(int j=0;j<x.size();j++){
New.push_back(0);
}
built.push_back(New);
}
int check=0;
for(int i=0;i<x.size();i++){
for(int j=0;j<x.size();j++){
if (x[i][j]) check=Union(i,j);
}
}
int New_Boss[1005];
for(int i=0;i<1005;i++){
New_Boss[i]=query(i);
}
//renew Union set;
for(int i=0;i<1005;i++){
Boss[i]=i;
Rank[i]=0;
}
vector<int> under_going={0};
for(int i=0;i<(int)x.size();i++){
for(int j=0;j<(int)x.size();j++){
if(x[i][j]==1) Union(i,j);
}
}
int now_Boss=New_Boss[0];
int Mark[1005]={0};
vector<int> tmp_List={0};
vector<int> cycle[1005];
for(int Time=0;Time<(int)x.size();Time++){
if(!Mark[Time]){
now_Boss=query(Time);
cycle[New_Boss[Time]].push_back(Time);
Mark[Time]=1;
for(int i=0;i<(int)x.size();i++){
if(now_Boss==query(i)&&Time!=i&&!Mark[i]){
//must=1
Mark[i]=1;
if(query(i)==now_Boss){
built[i][now_Boss]=1;
built[now_Boss][i]=1;
}
}
}
}
}
for(int i=0;i<1005;i++){
if((int)cycle[i].size()>2){
for(int j=0;j<(int)cycle[i].size()-1;j++){
built[cycle[i][j]][cycle[i][j+1]]=1;
built[cycle[i][j+1]][cycle[i][j]]=1;
}
built[cycle[i][0]][cycle[i][(int)cycle[i].size()-1]]=1;
built[cycle[i][(int)cycle[i].size()-1]][cycle[i][0]]=1;
}
}
return build(built);
}