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 <algorithm>
#include <bitset>
#include <cassert>
#include <chrono>
#include <complex>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <deque>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <ratio>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
#include <climits>
#include "supertrees.h"
#define ll long long
#define ld long double
#define mp make_pair
#define pb push_back
#define in insert
#define vll vector<ll>
#define endl "\n"
#define f first
#define s second
#define pll pair<int,int>
// #define int ll
using namespace std;
struct DSU
{
int connected;
vector<int> par, sz;
void init(int n)
{
par = sz = vector<int> (n + 1, 0);
for(int i = 1; i <= n; i++)
par[i] = i, sz[i] = 1;
connected = n;
}
int getPar(int u)
{
while(u != par[u])
{
par[u] = par[par[u]];
u = par[u];
}
return u;
}
int getSize(int u)
{
return sz[getPar(u)];
}
void unite(int u, int v)
{
int par1 = getPar(u), par2 = getPar(v);
if(par1 == par2)
return;
connected--;
if(sz[par1] > sz[par2])
swap(par1, par2);
sz[par2] += sz[par1];
sz[par1] = 0;
par[par1] = par[par2];
}
};
int construct(std::vector<std::vector<int>> p) {
DSU g,g2;
int n = p.size();
g.init(n-1);
g2.init(n-1);
std::vector<std::vector<int>> ans;
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 =i+1;j<n;j++){
if(p[i][j] == 3) return 0;
if(p[i][j] >= 1){
g.unite(i,j);
}
if(p[i][j] == 1){
g2.unite(i,j);
}
}
}
map<int,int> col;
vector<int> wow[n+1];
int c = 1;
for(int i =0;i<n;i++){
int op = g.getPar(i);
if(col[op] == 0){
col[op] = c;
c++;
}
wow[col[op]].pb(i);
}
c--;
for(int i =0;i<n;i++){
for(int j =0;j<n;j++){
ans[i][j] = 0;
}
}
for(int i = 1;i<=c;i++){
if(wow[i].size() == 1) continue;
set<int> contains;
bool ok = true;
for(int j = 0;j<wow[i].size();j++){
for(int k =j+1;k<wow[i].size();k++){
contains.in(p[wow[i][j]][wow[i][k]]);
if(p[wow[i][j]][wow[i][k]] == 0) return 0;
for(int l =0;l<wow[i].size();l++){
if(l == k || l == j) continue;
if(p[wow[i][j]][wow[i][k]] == 2 && (p[wow[i][j]][wow[i][l]] == 1 || p[wow[i][l]][wow[i][k]] == 1))
return 0;
}
}
}
if(contains.size() == 1 && *(contains.begin()) == 2){
if(wow[i].size() == 2) return 0;
}
col.clear();
int gg = 1;
vector<int> chains[n+2];
for(int j = 0;j<wow[i].size();j++){
int op = g2.getPar(wow[i][j]);
if(col[op] == 0){
col[op] = gg;
gg++;
}
chains[col[op]].pb(wow[i][j]);
}
vector<int> v;
for(int j =1;j<gg;j++){
// if(!poss) return 0;
for(int k=0;k<chains[j].size()-1;k++){
ans[chains[j][k]][chains[j][k+1]] = 1;
ans[chains[j][k+1]][chains[j][k]] = 1;
}
v.pb(chains[j].back());
}
for(int i= 0;i<v.size()-1;i++){
ans[v[i]][v[i+1]] = 1;
ans[v[i+1]][v[i]] = 1;
}
if(v.size() == 1) continue;
ans[v.back()][v[0]] = 1;
ans[v[0]][v.back()] = 1;
}
build(ans);
return 1;
}
// void solve(){
// }
// signed main(){
// ios_base::sync_with_stdio(0);
// cin.tie(NULL);
// // freopen(".in","r",stdin);freopen(".out","w",stdout);
// ll tt=1;
// // cin >> tt;
// while(tt--){
// solve();
// }
// }
Compilation message (stderr)
supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:142:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
142 | for(int j = 0;j<wow[i].size();j++){
| ~^~~~~~~~~~~~~~
supertrees.cpp:144:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
144 | for(int k =j+1;k<wow[i].size();k++){
| ~^~~~~~~~~~~~~~
supertrees.cpp:147:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
147 | for(int l =0;l<wow[i].size();l++){
| ~^~~~~~~~~~~~~~
supertrees.cpp:160:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
160 | for(int j = 0;j<wow[i].size();j++){
| ~^~~~~~~~~~~~~~
supertrees.cpp:172:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
172 | for(int k=0;k<chains[j].size()-1;k++){
| ~^~~~~~~~~~~~~~~~~~~
supertrees.cpp:179:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
179 | for(int i= 0;i<v.size()-1;i++){
| ~^~~~~~~~~~~
supertrees.cpp:141:8: warning: unused variable 'ok' [-Wunused-variable]
141 | bool ok = true;
| ^~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |