/*
In the name of Allah
made by: Ahmed_Solyman
*/
#include <bits/stdc++.h>
#include <ext/rope>
using namespace std;
using namespace __gnu_cxx;
#pragma GCC optimize("-Ofast")
#pragma GCC optimize("-O1")
//-------------------------------------------------------------//
typedef long long ll;
typedef unsigned long long ull;
#define fast ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define PI acos(-1)
#define lb lower_bound
#define ub upper_bound
#define all(v) v.begin(),v.end()
#define allr(v) v.rbegin(),v.rend()
#define sum_to(n) (n*(n+1))/2
#define pb push_back
#define pf push_front
#define fil(arr,x) memset(arr,x,sizeof(arr))
#define endl '\n'
const ll mod=1e9+7;
int dx[8]={0,1,0,-1,1,1,-1,-1};
int dy[8]={1,0,-1,0,1,-1,-1,1};
//-------------------------------------------------------------//
ll lcm(ll a,ll b)
{
return (max(a,b)/__gcd(a,b))*min(a,b);
}
void person_bool(bool x)
{
cout<<(x?"YES":"NO")<<endl;
}
void build(vector<vector<int>>b);
struct DSU{
vector<int>P,sz;
void init(int n){
P.resize(n+5);
sz.assign(n+5,1);
for(int i=0;i<n+5;i++)P[i]=i;
}
int find(int node){
if(P[node]==node)return node;
return P[node]=find(P[node]);
}
bool unite(int a,int b){
int p1=find(a);
int p2=find(b);
if(p1==p2){
return 0;
}
if(sz[p1]>sz[p2]){
swap(p1,p2);
}
P[p1]=p2;
sz[p2]+=sz[p1];
return 1;
}
}dsu,dsu2;
int construct(vector<vector<int>> p){
int n=p.size();
dsu.init(n);
dsu2.init(n);
set<int>s;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
s.insert(p[i][j]);
}
}
if((s.size()==2 && s.find(1)!=s.end() && s.find(2)!=s.end()) && s.size()==3){
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(p[i][j]==1)dsu.unite(i,j);
if(p[i][j]==2)dsu2.unite(i,j);
}
}
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(!p[i][j] && (dsu.find(i)==dsu.find(j) || dsu2.find(i)==dsu2.find(j))){
return 0;
}
if(dsu.find(i)==dsu2.find(i)){
return 0;
}
}
}
vector<int>v[n];
for(int i=0;i<n;i++)v[dsu.find(i)].push_back(i);
vector<vector<int>>b(n,vector<int>(n));
for(int i=0;i<n;i++){
int m=(int)v[i].size();
for(int j=1;j<m;j++){
b[v[i][j-1]][v[i][j]]=1;
b[v[i][j]][v[i][j-1]]=1;
}
}
for(int i=0;i<n;i++){
v[i].clear();
}
for(int i=0;i<n;i++)v[dsu2.find(i)].push_back(i);
Dsu d=dsu1;
for(int i=0;i<n;i++){
int m=(int)v[i].size();
if(m<=1)continue;
for(int j=1;j<m;j++){
b[v[i][j-1]][v[i][j]]=1;
b[v[i][j]][v[i][j-1]]=1;
dsu.unite(v[i][j-1],v[i][j]);
}
if(m==2){
return 0;
}
b[v[i][0]][v[i][m-1]]=1;
b[v[i][m-1]][v[i][0]]=1;
dsu.unite(v[i][0],v[i][m-1]);
}
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(dsu.find(i) == dsu.find(j) && p[i][j]==0){
return 0;
}
if((dsu.find(i) == dsu.find(j) && d.find(i) != d.find(j)) && p[i][j]==1){
return 0;
}
}
}
build(b);
return 1;
}
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(p[i][j])dsu.unite(i,j);
}
}
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(!p[i][j] && dsu.find(i)==dsu.find(j)){
return 0;
}
}
}
vector<int>v[n];
for(int i=0;i<n;i++)v[dsu.find(i)].push_back(i);
vector<vector<int>>b(n,vector<int>(n));
for(int i=0;i<n;i++){
bool cyc=1;
int m=(int)v[i].size();
for(int j=1;j<m;j++){
b[v[i][j-1]][v[i][j]]=1;
b[v[i][j]][v[i][j-1]]=1;
cyc&=(p[v[i][j]][v[i][j-1]]==2);
}
if(cyc){
if(m>2){
b[v[i][m-1]][v[i][0]]=1;
b[v[i][0]][v[i][m-1]]=1;
}
else if(m==2){
return 0;
}
}
}
build(b);
return 1;
}
Compilation message
supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:105:9: error: 'Dsu' was not declared in this scope; did you mean 'dsu'?
105 | Dsu d=dsu1;
| ^~~
| dsu
supertrees.cpp:126:51: error: 'd' was not declared in this scope
126 | if((dsu.find(i) == dsu.find(j) && d.find(i) != d.find(j)) && p[i][j]==1){
| ^
