#include "supertrees.h"
#include <vector>
#include<bits/stdc++.h>
using namespace std;
int construct(std::vector<std::vector<int>> p) {
int n = p.size();
std::vector<std::vector<int>> answer(n, vector<int>(n));
vector<int> par(n); iota(par.begin(), par.end(), 0);
function<int(int)> fset = [&] (int x) {return (par[x] == x ? x : par[x] = fset(par[x]));};
auto mg = [&] (int x, int y) {
if ((x = fset(x)) == (y = fset(y))) return;
par[y] = x;
};
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (p[i][j]) mg(i, j);
}
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (fset(i) == fset(j) && !p[i][j]) return 0;
}
}
for (int i = 0; i < n; i++) {
if (fset(i) != i) continue;
vector<int> c;
for (int j = 0; j < n; j++) {
if (fset(j) == i) c.emplace_back(j);
}
for (int j = 0; j < c.size() - 1; j++) {
answer[c[j]][c[j + 1]] = answer[c[j + 1]][c[j]] = 1;
}
}
build(answer);
return 1;
}
