# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
744724 | josanneo22 | Connecting Supertrees (IOI20_supertrees) | 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 <bits/stdc++.h>
#include<unordered_map>
#include<unordered_set>
#include<algorithm>
using namespace std;
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define fi first
#define se second
#include "supertrees.h"
#include <vector>
void build(vector<vector<int>> b);
const int N = 1005;
int n, mx, vis[N];
vector<int> block, edge, circle;
vector<vector<int>> path, ans;
void st(int u, int v) {
ans[u][v] = 1; ans[v][u] = 1;
}
void dfs(int u) {
vis[u] = 1;
block.push_back(u);
for (int v = 0; v < n; v++) {
mx = max(mx, path[u][v]);
if (!vis[v] && path[u][v]) dfs(v);
}
}
void find_cycle(int u) {
vis[u] = 2;
edge.push_back(u);
for (int v = 0; v < n; v++) {
if (vis[v] == 1 && path[u][v] == 1) find_cycle(v);
}
}
int construct(std::vector<std::vector<int>> p) {
path = p; n = p.size(); ans.resize(n);
for (int i = 0; i < n; i++) ans[i].resize(n);
for (int i = 0; i < n; i++) {
if (vis[i]) continue;
mx = 1; block.clear(); dfs(i);
if (mx == 3) return 0;
int sz = block.size();
for (int f = 1; f < sz; f++) {
for (int k = 0; k < f; k++) {
if (!path[block[j]][block[k]]) return 0;
}
}
if (mx == 1) {
for(int j=1;j<sz;j++)st(block[0], block[j]);
}
else {
circle.clear();
for (int j = 0, k = block[j]; j < sz; j++, k = block[j]) {
if (vis[k] != 1) continue;
edge.clear();
find_cycle(k);
for (int a = 1; a < edge.size(); a++) {
for (int b = 0; b < a; b++) {
if (path[edge[a]][edge[b]] != 1) return 0;
}
st(edge[0], edge[a]);
}
circle.push_back(edge[0]);
}
if (circle.size() <= 2) return 0;
int sz3 = circle.size();
for (int j = 0; j < sz3; j++) st(circle[j], circle[(j + 1) % sz3]);
}
}
build(ans);
return 1;
}