#include "parks.h"
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2e5 + 10;
vector<int> adj[MAXN];
void dfs(int node, vector<int> &vis)
{
vis[node] = 1;
for(auto a:adj[node])
if(!vis[node])
dfs(a, vis);
}
int construct_roads(vector<int> x, vector<int> y)
{
int n = x.size();
map<pair<int, int>, int> ft;
for(int i=0;i<n;i++)
{
ft[{x[i], y[i]}] = i;
}
vector<pair<int, int>> dir{{2, 0}, {-2, 0}, {0, 2}, {0, -2}};
vector<int> u, v, a, b;
for(int i=0;i<n;i++)
{
for(auto d:dir)
{
int cx = x[i] + d.first, cy = y[i] + d.second;
if(!ft.count({cx, cy}))
continue;
adj[i].push_back(ft[{cx, cy}]);
if(ft[{cx, cy}] > i)
{
u.push_back(i);
v.push_back(ft[{cx, cy}]);
if(d.second == 0)
{
a.push_back(x[i] + cx >> 1);
b.push_back(cy - 1);
}
else
{
if(x[i] == 2)
{
a.push_back(x[i] - 1);
b.push_back(y[i] + cy >> 1);
}
else
{
a.push_back(x[i] + 1);
b.push_back(y[i] + cy >> 1);
}
}
}
}
}
vector<int> vis(n, 0);
dfs(0, vis);
for(int i=0;i<n;i++)
if(!vis[i])
return 0;
build(u, v, a, b);
return 1;
}
Compilation message
parks.cpp: In function 'int construct_roads(std::vector<int>, std::vector<int>)':
parks.cpp:37:38: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
37 | a.push_back(x[i] + cx >> 1);
parks.cpp:45:42: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
45 | b.push_back(y[i] + cy >> 1);
parks.cpp:50:42: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
50 | b.push_back(y[i] + cy >> 1);
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Incorrect |
3 ms |
4948 KB |
Solution announced impossible, but it is possible. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Incorrect |
3 ms |
4948 KB |
Solution announced impossible, but it is possible. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Incorrect |
3 ms |
4948 KB |
Solution announced impossible, but it is possible. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Incorrect |
3 ms |
4948 KB |
Solution announced impossible, but it is possible. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Incorrect |
3 ms |
4948 KB |
Solution announced impossible, but it is possible. |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Incorrect |
3 ms |
4948 KB |
Solution announced impossible, but it is possible. |
3 |
Halted |
0 ms |
0 KB |
- |