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>
using namespace std;
using ll = long long;
#ifdef U_U
#include "algo/debug.h"
#else
#define deb(x...) 42
#endif
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define pb push_back
#define szof(x) (int)x.size()
#define int ll
#define love bool
template<class T> void MIN(T& a, const T b) { b < a ? a = b, 1 : 0; }
template<class T> void MAX(T& a, const T b) { a < b ? a = b, 1 : 0; }
const int mod = 1e9 + 7;
const int N = 2e5 + 5;
const int inf = 2147483647;
const int INF = 9223372036854775807;
const int dx[] = {1, -1, 0, 0};
const int dy[] = {0, 0, 1, -1};
// Follow the white rabbit.
int n, w, h;
char e[505][505];
int dist[505][505];
bool check (int i, int j) {
if (i > h || i < 1 || j > w || j < 1 || e[i][j] == 'x') return 0;
return 1;
}
void ka () {
map <int, int> left, right;
left[0] = 3;
left[1] = 2;
left[2] = 0;
left[3] = 1;
right[0] = 2;
right[1] = 3;
right[2] = 1;
right[3] = 0;
cin >> n >> w >> h;
vector <pair <int, int> > v;
for (int i = 1; i <= h; i ++) {
for (int j = 1; j <= w; j ++) {
cin >> e[i][j];
if (e[i][j] >= '1' && e[i][j] <= '9') v.pb({i, j});
}
}
int ans = inf;
for (int i = 1; i <= h; i ++) {
for (int lolx = 0; lolx < 2; lolx ++) {
int j = (lolx ? 1 : n);
if (e[i][j] == 'x') continue;
int lol = 0;
for (pair <int, int> c : v) {
queue <pair <int, int> > q;
q.push(c);
for (int I = 1; I <= h; I ++) for (int J = 1; J <= w; J ++) dist[I][J] = inf;
dist[c.first][c.second] = 0;
while (szof(q)) {
pair <int, int> tmp = q.front(); q.pop();
for (int k = 0; k < 4; k ++) {
int x = tmp.first, y = tmp.second;
int cur = k;
while (1) {
if (e[x][y] == 'A') cur = left[cur];
if(e[x][y] == 'C') cur = right[cur];
int X = x + dx[cur], Y = y + dy[cur];
if (!check(X, Y)) {
if (dist[x][y] > dist[tmp.first][tmp.second] + 1) {
dist[x][y] = dist[tmp.first][tmp.second] + 1;
q.push({x, y});
}
break;
}
x = X, y = Y;
}
}
}
lol += dist[i][j];
}
MIN(ans, lol);
}
}
for (int j = 1; j <= w; j ++) {
for (int lolx = 0; lolx < 2; lolx ++) {
int i = (lolx ? 1 : h);
if (e[i][j] == 'x') continue;
int lol = 0;
for (pair <int, int> c : v) {
queue <pair <int, int> > q;
q.push(c);
for (int I = 1; I <= h; I ++) for (int J = 1; J <= w; J ++) dist[I][J] = inf;
dist[c.first][c.second] = 0;
while (szof(q)) {
pair <int, int> tmp = q.front(); q.pop();
for (int k = 0; k < 4; k ++) {
int x = tmp.first, y = tmp.second;
int cur = k;
while (1) {
if (e[x][y] == 'A') cur = left[cur];
if(e[x][y] == 'C') cur = right[cur];
int X = x + dx[cur], Y = y + dy[cur];
if (!check(X, Y)) {
if (dist[x][y] > dist[tmp.first][tmp.second] + 1) {
dist[x][y] = dist[tmp.first][tmp.second] + 1;
q.push({x, y});
}
break;
}
x = X, y = Y;
}
}
}
lol += dist[i][j];
}
MIN(ans, lol);
}
}
cout << (ans == inf ? -1 : ans);
}
const bool overflow = 0;
main() {
//freopen("haircut.in" , "r", stdin); freopen("haircut.out", "w", stdout);
cout.setf(ios::fixed);cout.precision(12);
ios_base::sync_with_stdio(0);
cin.tie(0); cout.tie(0);
int tc = 1;
if (overflow) cin >> tc;
for (int cs = 1; cs <= tc; cs ++) {
#ifdef U_U
cout << ":D\n";
#endif
ka();
//if (cs != tc) cout << '\n';
}
}
Compilation message (stderr)
robots.cpp:130:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
130 | main() {
| ^~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |