# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
734877 | Elvin_Fritl | Comparing Plants (IOI20_plants) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "supertrees.h"
#include <bits/stdc++.h>
using namespace std;
bool vis[1002];
int construct(vector<vector<int> > p) {
int n = (int)p.size();
vector<vector<int> > ans(n);
for(int i=0;i<n;i++) ans[i].resize(n);
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(p[i][j] == 3) return 0;
}
}
for(int i=0;i<n;i++){
if(vis[i]) continue;
vis[i] = 1;
vector<int> x;
vector<vector<int> > y;
queue<int> q;
q.push(i);
while(!q.empty()){
int u = q.front();
q.pop();
vis[u] = 1;
bool use = 0;
x.push_back(u);
vector<int> z;
for(int v=0;v<n;v++){
if(vis[v]) continue;
if(p[u][v] == 0) continue;
if(p[u][v] == 1){
ans[u][v] = 1;
ans[v][u] = 1;
vis[v] = 1;
z.push_back(v);
}
else if(!use){
use = 1;
q.push(v);
}
}
y.push_back(z);
}
for(int i=0;i<y.size();i++){
for(int j=0;j<y[i].size();j++){
for(int k=j+1;k<y[i].size();k++){
if(p[y[i][j]][y[i][k]] != 1) return 0;
}
}
for(int j=0;j<x.size();j++){
if(i == j) continue;
for(int k=0;k<y[i].size();k++){
if(p[x[j]][y[i][k]] != 2) return 0;
}
}
for(int j=0;j<y.size();j++){
if(i == j) continue;
for(int a=0;a<y[i].size();a++){
for(int b=0;b<y[j].size();b++){
if(p[y[i][a]][y[j][b]] != 2) return 0;
}
}
}
}
for(int i=0;i<x.size();i++){
for(int j=i+1;j<x.size();j++){
if(p[x[i]][x[j]] != 2) return 0;
}
}
if((int)x.size() == 2) return 0;
else if((int)x.size() > 2){
for(int i=0;i<(int)x.size()-1;i++){
ans[x[i]][x[i+1]] = 1;
ans[x[i+1]][x[i]] = 1;
}
ans[0][x[(int)x.size()-1]] = 1;
ans[x[(int)x.size()-1]][0] = 1;
}
}
build(ans);
return 1;
}