#include "parks.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
struct Coord {
int x, y, i;
bool operator<(Coord other) const {
if (y == other.y) {
return x < other.x;
}
return y < other.y;
}
};
int n;
vector<Coord> coords;
std::vector<int> u, v, a, b;
vector<int> link;
int find(int x) {
if (link[x] != x) link[x] = find(link[x]);
return link[x];
}
void unite(int a, int b) {
int x = find(a), y = find(b);
link[x] = y;
}
bool same(int a, int b) {
return find(a) == find(b);
}
void connect(int i, int j) {
if (coords[i].x == coords[j].x) {
u.pb(coords[i].i);
v.pb(coords[j].i);
if (coords[i].x == 4)
a.pb(coords[i].x+1);
else
a.pb(coords[i].x-1);
b.pb(coords[i].y+1);
} else if (coords[i].y == coords[j].y) {
u.pb(coords[i].i);
v.pb(coords[j].i);
a.pb(coords[i].x+1);
b.pb(coords[i].y-1);
}
}
int construct_roads(std::vector<int> x, std::vector<int> y) {
if (x.size() == 1) {
build({}, {}, {}, {});
return 1;
}
n = x.size();
int maxX = 0;
for (int i = 0; i < n; i++) maxX = max(maxX, x[i]);
bool res = true;
if (maxX <= 4) {
coords = vector<Coord>(n);
for (int i = 0; i < n; i++) {
coords[i] = {x[i], y[i], i};
}
sort(coords.begin(), coords.end());
link = vector<int>(n);
iota(link.begin(), link.end(), 0);
for (int i = 0; i < n-1; i++) {
if ((coords[i].y+2 == coords[i+1].y && coords[i].x == coords[i+1].x) || (coords[i].y == coords[i+1].y && abs(coords[i+1].x-coords[i].x) == 2)) {
connect(i, i+1);
unite(i, i+1);
}
if (i+2 < n && ((coords[i].y+2 == coords[i+2].y && coords[i].x == coords[i+2].x) || (coords[i].y == coords[i+2].y && abs(coords[i+2].x-coords[i].x) == 2))) {
connect(i, i+2);
unite(i, i+2);
}
}
for (int i = 0; i < n-1; i++) {
res &= find(i) == find(i+1);
}
} else {
int maxY = 0;
for (int i = 0; i < n; i++) maxY = max(maxY, y[i]);
link = vector<int>(n);
iota(link.begin(), link.end(), 0);
vector<vector<int>> mat(maxY/2, vector<int>(3)), matIdx(maxY/2, vector<int>(3));
for (int i = 0; i < n; i++) {
mat[y[i]/2-1][x[i]/2-1] = 1;
matIdx[y[i]/2-1][x[i]/2-1] = i;
}
for (int i = 0; i < maxY/2; i++) {
if (mat[i][0] && mat[i][1]) {
u.pb(matIdx[i][0]);
v.pb(matIdx[i][1]);
a.pb(3);
b.pb(2*i+1);
unite(matIdx[i][0], matIdx[i][1]);
}
if (mat[i][1] && mat[i][2]) {
u.pb(matIdx[i][1]);
v.pb(matIdx[i][2]);
a.pb(5);
b.pb(2*i+1);
unite(matIdx[i][1], matIdx[i][2]);
}
if (i == maxY/2-1) continue;
bool isValid = false;
if (mat[i][0] && mat[i+1][0]) {
u.pb(matIdx[i][0]);
v.pb(matIdx[i+1][0]);
a.pb(1);
b.pb(2*i+3);
unite(matIdx[i][0], matIdx[i+1][0]);
isValid = true;
}
if (mat[i][2] && mat[i+1][2]) {
u.pb(matIdx[i][2]);
v.pb(matIdx[i+1][2]);
a.pb(7);
b.pb(2*i+3);
unite(matIdx[i][2], matIdx[i+1][2]);
isValid = true;
}
if (!isValid && mat[i][1] && mat[i+1][1]) {
u.pb(matIdx[i][1]);
v.pb(matIdx[i+1][1]);
a.pb(3);
b.pb(2*i+3);
unite(matIdx[i][1], matIdx[i+1][1]);
}
}
for (int i = 0; i < n-1; i++) {
res &= find(i) == find(i+1);
}
}
if (!res) return 0;
build(u, v, a, b);
return 1;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 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 |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
54 ms |
9396 KB |
Output is correct |
10 |
Correct |
5 ms |
1364 KB |
Output is correct |
11 |
Correct |
27 ms |
5200 KB |
Output is correct |
12 |
Correct |
7 ms |
1944 KB |
Output is correct |
13 |
Correct |
8 ms |
2892 KB |
Output is correct |
14 |
Correct |
0 ms |
340 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
53 ms |
9376 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 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 |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
54 ms |
9396 KB |
Output is correct |
10 |
Correct |
5 ms |
1364 KB |
Output is correct |
11 |
Correct |
27 ms |
5200 KB |
Output is correct |
12 |
Correct |
7 ms |
1944 KB |
Output is correct |
13 |
Correct |
8 ms |
2892 KB |
Output is correct |
14 |
Correct |
0 ms |
340 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
53 ms |
9376 KB |
Output is correct |
17 |
Correct |
0 ms |
212 KB |
Output is correct |
18 |
Correct |
0 ms |
212 KB |
Output is correct |
19 |
Correct |
0 ms |
212 KB |
Output is correct |
20 |
Correct |
0 ms |
212 KB |
Output is correct |
21 |
Correct |
0 ms |
212 KB |
Output is correct |
22 |
Correct |
0 ms |
212 KB |
Output is correct |
23 |
Correct |
141 ms |
23204 KB |
Output is correct |
24 |
Correct |
1 ms |
212 KB |
Output is correct |
25 |
Correct |
1 ms |
468 KB |
Output is correct |
26 |
Correct |
1 ms |
468 KB |
Output is correct |
27 |
Correct |
1 ms |
596 KB |
Output is correct |
28 |
Correct |
55 ms |
10136 KB |
Output is correct |
29 |
Correct |
83 ms |
14984 KB |
Output is correct |
30 |
Correct |
108 ms |
18636 KB |
Output is correct |
31 |
Correct |
140 ms |
24220 KB |
Output is correct |
32 |
Correct |
0 ms |
212 KB |
Output is correct |
33 |
Correct |
1 ms |
212 KB |
Output is correct |
34 |
Correct |
1 ms |
212 KB |
Output is correct |
35 |
Correct |
0 ms |
212 KB |
Output is correct |
36 |
Correct |
0 ms |
212 KB |
Output is correct |
37 |
Correct |
1 ms |
212 KB |
Output is correct |
38 |
Correct |
1 ms |
212 KB |
Output is correct |
39 |
Correct |
0 ms |
212 KB |
Output is correct |
40 |
Correct |
0 ms |
300 KB |
Output is correct |
41 |
Correct |
0 ms |
212 KB |
Output is correct |
42 |
Correct |
0 ms |
212 KB |
Output is correct |
43 |
Correct |
1 ms |
340 KB |
Output is correct |
44 |
Correct |
1 ms |
468 KB |
Output is correct |
45 |
Correct |
53 ms |
9732 KB |
Output is correct |
46 |
Correct |
78 ms |
13976 KB |
Output is correct |
47 |
Correct |
79 ms |
14104 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 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 |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
54 ms |
9396 KB |
Output is correct |
10 |
Correct |
5 ms |
1364 KB |
Output is correct |
11 |
Correct |
27 ms |
5200 KB |
Output is correct |
12 |
Correct |
7 ms |
1944 KB |
Output is correct |
13 |
Correct |
8 ms |
2892 KB |
Output is correct |
14 |
Correct |
0 ms |
340 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
53 ms |
9376 KB |
Output is correct |
17 |
Correct |
0 ms |
212 KB |
Output is correct |
18 |
Correct |
0 ms |
212 KB |
Output is correct |
19 |
Correct |
0 ms |
212 KB |
Output is correct |
20 |
Correct |
0 ms |
212 KB |
Output is correct |
21 |
Correct |
0 ms |
212 KB |
Output is correct |
22 |
Correct |
0 ms |
212 KB |
Output is correct |
23 |
Correct |
141 ms |
23204 KB |
Output is correct |
24 |
Correct |
1 ms |
212 KB |
Output is correct |
25 |
Correct |
1 ms |
468 KB |
Output is correct |
26 |
Correct |
1 ms |
468 KB |
Output is correct |
27 |
Correct |
1 ms |
596 KB |
Output is correct |
28 |
Correct |
55 ms |
10136 KB |
Output is correct |
29 |
Correct |
83 ms |
14984 KB |
Output is correct |
30 |
Correct |
108 ms |
18636 KB |
Output is correct |
31 |
Correct |
140 ms |
24220 KB |
Output is correct |
32 |
Correct |
0 ms |
212 KB |
Output is correct |
33 |
Correct |
1 ms |
212 KB |
Output is correct |
34 |
Correct |
1 ms |
212 KB |
Output is correct |
35 |
Correct |
0 ms |
212 KB |
Output is correct |
36 |
Correct |
0 ms |
212 KB |
Output is correct |
37 |
Correct |
1 ms |
212 KB |
Output is correct |
38 |
Correct |
1 ms |
212 KB |
Output is correct |
39 |
Correct |
0 ms |
212 KB |
Output is correct |
40 |
Correct |
0 ms |
300 KB |
Output is correct |
41 |
Correct |
0 ms |
212 KB |
Output is correct |
42 |
Correct |
0 ms |
212 KB |
Output is correct |
43 |
Correct |
1 ms |
340 KB |
Output is correct |
44 |
Correct |
1 ms |
468 KB |
Output is correct |
45 |
Correct |
53 ms |
9732 KB |
Output is correct |
46 |
Correct |
78 ms |
13976 KB |
Output is correct |
47 |
Correct |
79 ms |
14104 KB |
Output is correct |
48 |
Correct |
0 ms |
212 KB |
Output is correct |
49 |
Correct |
0 ms |
304 KB |
Output is correct |
50 |
Correct |
0 ms |
212 KB |
Output is correct |
51 |
Correct |
1 ms |
212 KB |
Output is correct |
52 |
Correct |
1 ms |
304 KB |
Output is correct |
53 |
Correct |
0 ms |
212 KB |
Output is correct |
54 |
Incorrect |
1 ms |
212 KB |
Tree @(3, 5) appears more than once: for edges on positions 3 and 4 |
55 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 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 |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
54 ms |
9396 KB |
Output is correct |
10 |
Correct |
5 ms |
1364 KB |
Output is correct |
11 |
Correct |
27 ms |
5200 KB |
Output is correct |
12 |
Correct |
7 ms |
1944 KB |
Output is correct |
13 |
Correct |
8 ms |
2892 KB |
Output is correct |
14 |
Correct |
0 ms |
340 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
53 ms |
9376 KB |
Output is correct |
17 |
Runtime error |
1 ms |
340 KB |
Execution killed with signal 11 |
18 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 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 |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
54 ms |
9396 KB |
Output is correct |
10 |
Correct |
5 ms |
1364 KB |
Output is correct |
11 |
Correct |
27 ms |
5200 KB |
Output is correct |
12 |
Correct |
7 ms |
1944 KB |
Output is correct |
13 |
Correct |
8 ms |
2892 KB |
Output is correct |
14 |
Correct |
0 ms |
340 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
53 ms |
9376 KB |
Output is correct |
17 |
Runtime error |
45 ms |
19420 KB |
Execution killed with signal 11 |
18 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 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 |
0 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
0 ms |
212 KB |
Output is correct |
9 |
Correct |
54 ms |
9396 KB |
Output is correct |
10 |
Correct |
5 ms |
1364 KB |
Output is correct |
11 |
Correct |
27 ms |
5200 KB |
Output is correct |
12 |
Correct |
7 ms |
1944 KB |
Output is correct |
13 |
Correct |
8 ms |
2892 KB |
Output is correct |
14 |
Correct |
0 ms |
340 KB |
Output is correct |
15 |
Correct |
1 ms |
340 KB |
Output is correct |
16 |
Correct |
53 ms |
9376 KB |
Output is correct |
17 |
Correct |
0 ms |
212 KB |
Output is correct |
18 |
Correct |
0 ms |
212 KB |
Output is correct |
19 |
Correct |
0 ms |
212 KB |
Output is correct |
20 |
Correct |
0 ms |
212 KB |
Output is correct |
21 |
Correct |
0 ms |
212 KB |
Output is correct |
22 |
Correct |
0 ms |
212 KB |
Output is correct |
23 |
Correct |
141 ms |
23204 KB |
Output is correct |
24 |
Correct |
1 ms |
212 KB |
Output is correct |
25 |
Correct |
1 ms |
468 KB |
Output is correct |
26 |
Correct |
1 ms |
468 KB |
Output is correct |
27 |
Correct |
1 ms |
596 KB |
Output is correct |
28 |
Correct |
55 ms |
10136 KB |
Output is correct |
29 |
Correct |
83 ms |
14984 KB |
Output is correct |
30 |
Correct |
108 ms |
18636 KB |
Output is correct |
31 |
Correct |
140 ms |
24220 KB |
Output is correct |
32 |
Correct |
0 ms |
212 KB |
Output is correct |
33 |
Correct |
1 ms |
212 KB |
Output is correct |
34 |
Correct |
1 ms |
212 KB |
Output is correct |
35 |
Correct |
0 ms |
212 KB |
Output is correct |
36 |
Correct |
0 ms |
212 KB |
Output is correct |
37 |
Correct |
1 ms |
212 KB |
Output is correct |
38 |
Correct |
1 ms |
212 KB |
Output is correct |
39 |
Correct |
0 ms |
212 KB |
Output is correct |
40 |
Correct |
0 ms |
300 KB |
Output is correct |
41 |
Correct |
0 ms |
212 KB |
Output is correct |
42 |
Correct |
0 ms |
212 KB |
Output is correct |
43 |
Correct |
1 ms |
340 KB |
Output is correct |
44 |
Correct |
1 ms |
468 KB |
Output is correct |
45 |
Correct |
53 ms |
9732 KB |
Output is correct |
46 |
Correct |
78 ms |
13976 KB |
Output is correct |
47 |
Correct |
79 ms |
14104 KB |
Output is correct |
48 |
Correct |
0 ms |
212 KB |
Output is correct |
49 |
Correct |
0 ms |
304 KB |
Output is correct |
50 |
Correct |
0 ms |
212 KB |
Output is correct |
51 |
Correct |
1 ms |
212 KB |
Output is correct |
52 |
Correct |
1 ms |
304 KB |
Output is correct |
53 |
Correct |
0 ms |
212 KB |
Output is correct |
54 |
Incorrect |
1 ms |
212 KB |
Tree @(3, 5) appears more than once: for edges on positions 3 and 4 |
55 |
Halted |
0 ms |
0 KB |
- |