#include "supertrees.h"
#include <vector>
#include <iostream>
#define ll long long
#define pb push_back
using namespace std;
const int N = 20005;
vector<vector<int>> ans;
int d[N], used[N];
int get_parent(int v){
if(v == d[v])
return v;
return d[v] = get_parent(d[v]);
}
void add(int u, int v){
int f = get_parent(u), s = get_parent(v);
if(f != s)
d[f] = d[s];
}
int construct(std::vector<std::vector<int>> p) {
int n = p.size();
int fault = 0;
for(int i = 0; i < n; i++)
d[i] = i;
for (int i = 0; i < n; i++) {
std::vector<int> row;
row.resize(n);
ans.push_back(row);
}
for(int i = 0; i < n; i++){
for(int j = 0; j < n; j++){
if(p[i][j]){
add(i, j);
}
}
}
int pi, pj;
for(int i = 0; i < n; i++){
for(int j = 0; j < n; j++){
pi = get_parent(i);
pj = get_parent(j);
if(p[i][j]){
if(pi != pj)
return 0;
if(pi != i)
ans[i][pi] = ans[pi][i] = 1;
if(pi != j)
ans[j][pi] = ans[pi][j] = 1;
}
else{
if(pi == pj)
return 0;
}
}
}
for(int i = 0; i < n; i++){
for(int j = 0; j < n; j++){
if(p[i][j] == 2){
pi = get_parent(i);
if(used[pi] || pi == i || pi == j)
continue;
ans[i][j] = ans[j][i] = 1;
used[pi] = 1;
}
}
}
for(int i = 0; i < n; i++){
for(int j = 0; j < n; j++){
if(p[i][j] == 2){
pi = get_parent(i);
if(!used[pi])
return 0;
}
}
}
build(ans);
return 1;
}
/*
5
1 1 0 0 0
1 1 0 0 0
0 0 1 2 2
0 0 2 1 2
0 0 2 2 1
*/
Compilation message
plants.cpp:1:10: fatal error: supertrees.h: No such file or directory
1 | #include "supertrees.h"
| ^~~~~~~~~~~~~~
compilation terminated.
