#include "supertrees.h"
#include <bits/stdc++.h>
using namespace std;
struct setunion{
int N;
vector<int> p, s;
setunion(int n) {
N = n;
p = vector<int>(N, -1);
s = vector<int>(N, 1);
}
int root(int n) {
int i = n;
while (p[i] != -1)
i = p[i];
return i;
}
void combine(int a, int b) {
int ra = root(a), rb = root(b);
if (ra == rb) return;
if (s[ra] < s[rb]) swap(ra, rb);
s[ra] += s[rb];
p[rb] = ra;
}
vector<vector<int>> extract() {
vector<vector<int>> erg(N), rem;
for (int i = 0; i < N; i++)
{
erg[root(i)].push_back(i);
}
for (vector<int> e: erg)
if (e.size())
rem.push_back(e);
return rem;
}
};
int construct(vector<vector<int>> p) {
int N = p.size();
int c = 0, o = 0;
setunion zsm(N);
for (int i = 0; i < N; i++)
{
for (int j = 0; j < N; j++)
{
if (p[i][j] == 3) return 0;
if (p[i][j]) zsm.combine(i, j);
if (p[i][j] || i == j) o++;
if (p[i][j] || i == j) c++;
}
}
vector<vector<int>> graph = vector<vector<int>>(N, vector<int>(N));
for (vector<int> z: zsm.extract()) {
c -= z.size() * z.size();
setunion one(z.size());
for (int i = 0; i < z.size(); ++i)
{
for (int j = 0; j < z.size(); ++j)
{
if (p[i][j] == 1)
one.combine(i, j);
}
}
vector<vector<int>> g = one.extract();
for (int i = 0; i < g.size(); i++)
{
graph[z[g[i][0]]][z[g[(i + 1) % g.size()][0]]] = true;
graph[z[g[(i + 1) % g.size()][0]]][z[g[i][0]]] = true;
}
for (int i = 0; i < g.size(); i++)
{
o -= g[i].size() * g[i].size();
for (int j = 1; j < g[i].size(); j++)
{
graph[z[g[i][0]]][z[g[i][j]]] = true;
}
}
}
for (int i = 0; i < N; i++)
graph[i][i] = false;
if (c || o) return 0;
build(graph);
return 1;
}
Compilation message
supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:69:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
69 | for (int i = 0; i < z.size(); ++i)
| ~~^~~~~~~~~~
supertrees.cpp:71:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
71 | for (int j = 0; j < z.size(); ++j)
| ~~^~~~~~~~~~
supertrees.cpp:80:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
80 | for (int i = 0; i < g.size(); i++)
| ~~^~~~~~~~~~
supertrees.cpp:86:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
86 | for (int i = 0; i < g.size(); i++)
| ~~^~~~~~~~~~
supertrees.cpp:89:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
89 | for (int j = 1; j < g[i].size(); j++)
| ~~^~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Incorrect |
0 ms |
212 KB |
b is not symmetric: b[0][1] (1) != b[1][0] (0) |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Incorrect |
0 ms |
212 KB |
b is not symmetric: b[0][1] (1) != b[1][0] (0) |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Incorrect |
0 ms |
212 KB |
Answer gives possible 0 while actual possible 1 |
6 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Incorrect |
1 ms |
212 KB |
Answer gives possible 0 while actual possible 1 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Incorrect |
0 ms |
212 KB |
b is not symmetric: b[0][1] (1) != b[1][0] (0) |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Incorrect |
0 ms |
212 KB |
b is not symmetric: b[0][1] (1) != b[1][0] (0) |
3 |
Halted |
0 ms |
0 KB |
- |