#include "supertrees.h"
#include <vector>
#include <bits/stdc++.h>
using namespace std;
//#define int long long
typedef vector<int> vi;
typedef pair<int,int> pii;
#define fi first
#define se second
#define sz(x) (int)(x).size()
#define debug(x) cerr<<#x<<" is "<<x<<endl;
#define debugl(x) cerr<<#x<<" is ";for(auto p:x)cerr<<p<<' ';cerr<<endl;
#define pb push_back
int construct(std::vector<std::vector<int>> p) {
int n = p.size();
vector<vi> t,c;
set<int> s,ps;
vector<vi> a(n);
for(int i=0;i<n;i++){
for(int j=0;j<n;j++)a[i].pb(0);
}
int par[n],cpar[n];
for(int i=0;i<n;i++){
if(s.find(i)!=s.end())continue;
t.pb({i});
par[i]=i;
ps.insert(i);//tree parents
for(int j=i+1;j<n;j++){
if(p[i][j]==1){
s.insert(j);
t.back().pb(j);
par[j]=i;
}
}
}
s.clear();
for(auto i:t){
int ct=i[0];
if(s.find(ct)!=s.end())continue;
c.pb({ct});
s.insert(ct);
cpar[ct]=ct;
for(auto j:ps){
if(p[ct][j]==2){
s.insert(j);
cpar[j]=ct;
c.back().pb(j);
}
}
}
for(auto v:t){
//debugl(v)
for(int i=1;i<sz(v);i++){
a[v[0]][v[i]]=1;
a[v[i]][v[0]]=1;
}
}
for(auto v:c){
//debugl(v)
if(sz(v)==1)continue;
if(sz(v)==2)return 0;
for(int i=0;i<sz(v)-1;i++){
a[v[i]][v[i+1]]=1;
a[v[i+1]][v[i]]=1;
}
a[v[0]][v.back()]=1;
a[v.back()][v[0]]=1;
}
//validate
for(auto v:t){
for(int i=0;i<sz(v);i++){
for(int j=i+1;j<sz(v);j++){
if(p[v[i]][v[j]]!=1)return 0;
}
}
}
for(auto v:c){
for(int i=0;i<sz(v);i++){
for(int j=i+1;j<sz(v);j++){
if(p[v[i]][v[j]]!=2)return 0;
}
}
}
for(int i=0;i<n;i++){
if(p[par[i]]!=p[i])return 0;
}
for(int i=0;i<n;i++){
if(par[i]!=i)continue;
for(int j=0;j<n;j++){
if(p[i][j]==1)p[i][j]=0;
if(p[cpar[i]][j]==1)p[cpar[i]][j]=0;
if(j==cpar[i]||i==j)continue;
if(p[cpar[i]][j]!=p[i][j])return 0;
}
//if(p[cpar[i]]!=p[i])return 0;
}
build(a);
return 1;
}
