//#pragma GCC optimize("O3") #include <bits/stdc++.h> #include "supertrees.h" using namespace std; const int sz=1001,INF=1000000000; vector<int>parent(sz,-1); int get(int num) { if(parent[num]<0) { return num; } else { parent[num]=get(parent[num]); return get(parent[num]); } } void unite(int num1,int num2) { int res1=get(num1),res2=get(num2); if(res1!=res2) { if(parent[res1]>parent[res2]) { swap(res1,res2); } parent[res1]+=parent[res2],parent[res2]=res1; } } int construct(vector<vector<int>>nums) { vector<vector<int>>res(nums.size()),ans(nums.size(),vector<int>(nums.size(),0)); for(int i=0;i<nums.size();i++) { for(int j=0;j<nums.size();j++) { if(nums[i][j]==2) { unite(i,j); } } } for(int i=0;i<nums.size();i++) { for(int j=0;j<nums.size();j++) { if(nums[i][j]==0) { if(get(i)==get(j)) { return 0; } } } } for(int i=0;i<nums.size();i++) { res[get(i)].push_back(i); } for(int i=0;i<nums.size();i++) { if(nums[i].size()==2) { return 0; } else { if(nums[i].size()!=1) { for(int j=0;j<nums[i].size();j++) { ans[nums[i][j]][nums[i][(j+1)%nums[i].size()]]=1,ans[nums[i][(j+1)%nums[i].size()]][nums[i][j]]=1; } } } } build(ans); return 1; }