# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
310560 | DS007 | 슈퍼트리 잇기 (IOI20_supertrees) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "supertrees.h"
using namespace std;
const int N = 1005;
class DSU {
int par[N], n;
public:
DSU(int n) {
this->n = n;
iota(par, par + n, 0);
}
int find(int x) {
if (x == par[x])
return x;
return par[x] = find(par[x]);
}
void merge(int u, int v) {
u = find(u), v = find(v);
par[u] = v;
}
};
int construct(vector<vector<int>> p) {
int n = p.size();
DSU dsu(n);
if (n == 1) {
build({{0}});
return 1;
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (p[i][j] == 3)
return 0;
else if (p[i][j] != 0)
dsu.merge(i, j);
}
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (p[i][j] == 0 && dsu.find(i) == dsu.find(j))
return 0;
}
}
vector<vector<int>> comp(n);
vector<vector<int>> b(n);
for (int i = 0; i < n; i++) b[i].resize(n, 0);
vector<bool> done(n, false);
for (int i = 0, k = 0; i < n; i++) {
if (done[i])
continue;
comp[k].push_back(i);
for (int j = i + 1; j < n; j++) {
if (dsu.find(i) == dsu.find(j))
comp[k].push_back(j), done[j] = true;
}
k++;
}
for (int i = 0; i < comp.size() && !comp[i].empty(); i++) {
//cout << "Done" << endl;
if (comp[i].size() == 1) continue;
set<int> s;
for (int j = 0; j < comp[i].size(); j++) {
for (int k = 0; k < comp[i].size(); k++) {
if (k != j && p[comp[i][j]][comp[i][k]] == 1)
s.insert(comp[i][j]), s.insert(comp[i][k]);
}
}
for (int j : s) {
for (int k : s) {
if (p[j][k] != 1)
return 0;
}
}
if (s.size() == comp[i].size() - 1)
return 0;
if (s.empty()) {
if (comp[i].size() <= 2)
return 0;
int last = -1;
for (int j : comp[i]) {
if (last != -1)
b[j][last] = b[last][j] = 1;
last = j;
}
b[*comp[i].begin()][*comp[i].rbegin()] = b[*comp[i].rbegin()][*comp[i].begin()] = 1;
}
int last = -1;
for (int j : s) {
if (last != -1)
b[j][last] = b[last][j] = 1;
last = j;
}
int last1 = -1, first1 = -1;
for (int j : comp[i]) {
if (last1 != -1)
b[last1][j] = b[j][last1] = 1;
else
first1 = j;
last1 = j;
}
if (s.size() != comp[i].size()) {
if (last != -1 && first1 != -1)
b[last][first1] = b[first1][last] = 1,
if (last != -1 && last1 != -1)
b[last][last1] = b[last1][last] = 1;
}
}
build(b);
return 1;
}