一只蚂蚁坐在由白色和黑色方格构成的无限网格上。开始时,网格全白,蚂蚁面向右侧。每行走一步,蚂蚁执行以下操作。
(1) 如果在白色方格上,则翻转方格的颜色,向右(顺时针)转 90 度,并向前移动一个单位。 (2) 如果在黑色方格上,则翻转方格的颜色,向左(逆时针方向)转 90 度,并向前移动一个单位。
编写程序来模拟蚂蚁执行的前 K 个动作,并返回最终的网格。
网格由数组表示,每个元素是一个字符串,代表网格中的一行,黑色方格由 'X' 表示,白色方格由 '_' 表示,蚂蚁所在的位置由 'L', 'U', 'R', 'D' 表示,分别表示蚂蚁 左、上、右、下 的朝向。只需要返回能够包含蚂蚁走过的所有方格的最小矩形。
示例 1:
输入: 0 输出: ["R"] 示例 2:
输入: 2 输出: [ "_X", "LX" ] 示例 3:
输入: 5 输出: [ "_U", "X_", "XX" ] 说明:
K <= 100000
思路:直接模拟蚂蚁行走过程即可,为了防止数组保内存,我们需要些简单操作。
class Solution { public List<String> printKMoves(int K) { char c = 'R'; int x = 100000, y = 100000; int mnx = 100000, mny = 100000, mxx = 100000, mxy = 100000; Map<String, Integer> map = new HashMap<>(); for (int i = 0; i < K; i++) { String s = String.valueOf(x) + "#" + String.valueOf(y); c = change(c, map.getOrDefault(s, 0) % 2); if (c == 'U') x--; else if (c == 'D') x++; else if (c == 'L') y--; else y++; mnx = Math.min(mnx, x); mxx = Math.max(mxx, x); mny = Math.min(mny, y); mxy = Math.max(mxy, y); map.put(s, map.getOrDefault(s, 0) + 1); } String ss = String.valueOf(x) + "#" + String.valueOf(y); map.put(ss, map.getOrDefault(ss, 0) + 1); int m = mxx - mnx + 1; int n = mxy - mny + 1; char[][] cc = new char[m][n]; for (int i = 0; i < m; i++) for (int j = 0; j < n; j++) { int xx = i + mnx; int yy = j + mny; if (xx == x && yy == y) cc[i][j] = c; else { String s = String.valueOf(xx) + "#" + String.valueOf(yy); if (map.containsKey(s)) cc[i][j] = map.get(s) % 2 == 0 ? '_' : 'X'; else cc[i][j] = '_'; } } List<String> ans = new ArrayList<>(); for (int i = 0; i < m; i++) ans.add(String.valueOf(cc[i])); return ans; } private char change(char c, int id) { if (c == 'L') return id == 0 ? 'U' : 'D'; if (c == 'R') return id == 0 ? 'D' : 'U'; if (c == 'U') return id == 0 ? 'R' : 'L'; return id == 0 ? 'L' : 'R'; } }