#include <iostream>
#include "supertrees.h"
using namespace std;
int rt[1<<10];
int construct(vector<vector<int>> p){
int n = p.size();
vector<vector<int>> adj(n, vector<int> (n, 0));
for (int i=0;i<n;i++){
if (rt[i] == 0){
rt[i] = i;
for (int j=0;j<n;j++)
if (i != j and p[i][j] == 1)
rt[j] = i, adj[i][j] = adj[j][i] = 1;
}
}
for (int i=0;i<n;i++){
int t = (i == rt[i]);
for (int j=i+1;t and j<n;j++){
if (p[i][j] == 2)
adj[i][rt[j]] = adj[rt[j]][i] = 1, t = 0;
}
for (int j=0;t and j < i;j++){
if (p[i][j] == 2)
adj[i][rt[j]] = adj[rt[j]][i] = 1, t = 0;
}
}
build(adj);
return 1;
}
