Submission #1002124

#	Time	Username	Problem	Language	Result	Execution time	Memory
1002124		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.
//     3
//   1   0
//     2
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 J = 1; J <= w; J ++) dist[I][J] = inf;
	for (int i = 1; i <= h; i ++) {
		for (int j = 1; j <= w; j ++) {
			int lol = 0;
			for (pair <int, int> c : v) {
				queue <pair <int, int> > q;
				q.push(c);
				dist[c.first][c.second] = 0;
				vector <pair <int, int> > del;
				while (szof(q)) {
					pair <int, int> tmp = q.front(); q.pop();
					del.pb(tmp);
					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];
				for (pair <int, int> t : del) dist[t.first][t.second] = inf;
			}
			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:98:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   98 | main() {
      | ^~~~

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