Submission #1002180

#	Time	Username	Problem	Language	Result	Execution time	Memory
1002180		vjudge1	Robots (APIO13_robots)	C++17	10 / 100	1 ms	348 KiB

robots

#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, h, w;
char e[505][505];
int dist[505][505][10];
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 >> h >> w;
	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});
		}
	}
	for (int i = 1; i <= h; i ++) for (int j = 1; j <= w; j ++) for (int id = 0; id < szof(v); id ++) dist[i][j][id] = inf;
	for (int id = 0; id < szof(v); id ++) {
		queue <pair <int, int> > q;
		q.push(v[id]);
		dist[v[id].first][v[id].second][id] = 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][id] > dist[tmp.first][tmp.second][id] + 1) {
							dist[x][y][id] = dist[tmp.first][tmp.second][id]+ 1;
							q.push({x, y});
						}
						break;
					}
					x = X, y = Y;
				}
			}
		}
	}
	int ans = inf;
	for (int i = 1; i <= h; i ++) {
		for (int j = 1; j <= w; j ++) {
			int lol = 0;
			for (int id = 0; id < szof(v); id ++) lol += dist[i][j][id];
			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:94:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   94 | main() {
      | ^~~~

• Subtask 1: 10 / 10
• Subtask 2: 0 / 20
• Subtask 3: 0 / 30
• Subtask 4: 0 / 40