SnakeBot simplify

src/SnakeBot.java

@@ -19,10 +19,14 @@ public class SnakeBot extends Bot {
     }
 
     @Override
-    protected char nextMove(View view) throws Exception {
+    protected char nextMove(View view) {
         System.out.println();
-        String data = view.data;
-        char[][] grid = dataToGrid(view.data, view.width);
+        String data = view.data
+                .replace('^', '*')
+                .replace('<', '*')
+                .replace('>', '*')
+                .replace('V', '*');
+        char[][] grid = dataToGrid(data, view.width);
 
         if (data.contains("@")) {
             return breadthFirstSearch(grid);
@@ -31,11 +35,10 @@ public class SnakeBot extends Bot {
         }
     }
 
-    protected char[][] dataToGrid(String data, int width) {
-        int height = data.length() / width;
-        char[][] grid = new char[width][height];
+    protected char[][] dataToGrid(String data, int size) {
+        char[][] grid = new char[size][size];
         for (int i = 0; i < data.length(); i++) {
-            grid[i % width][i / width] = data.charAt(i);
+            grid[i % size][i / size] = data.charAt(i);
         }
         return grid;
     }
@@ -58,34 +61,61 @@ public class SnakeBot extends Bot {
     protected char safeMove(char[][] grid) {
         int size = grid.length;
         if (grid[size / 2][size / 2 - 1] == '*') {
-            return '<';
+            return safeTurn(grid);
         } else {
             return '^';
         }
     }
 
+    protected char safeTurn(char[][] grid) {
+        int size = grid.length;
+
+        int leftStars = 0;
+        int rightStars = 0;
+        for (int x = 0; x < size / 2; x++) {
+            for (int y = 0; y < size; y++) {
+                if (grid[x][y] == '*')
+                    leftStars++;
+                if (grid[size - x - 1][y] == '*')
+                    rightStars++;
+            }
+        }
+
+        if (leftStars > rightStars) {
+            return '>';
+        } else {
+            return '<';
+        }
+    }
+
     protected char breadthFirstSearch(char[][] grid) {
-        int width = grid[0].length;
-        int height = grid.length;
-        int startX = width / 2;
-        int startY = height / 2;
+        int size = grid.length;
+        int start = size / 2;
 
+        boolean[][] visited = new boolean[size][size];
         Queue<Move> queue = new LinkedList<>();
-        queue.add(new Move(startX, startY - 1, '^'));
-        queue.add(new Move(startX, startY + 1, '>'));
-        queue.add(new Move(startX + 1, startY, '>'));
-        queue.add(new Move(startX - 1, startY, '<'));
+
+        int[][] directions = {{0, -1}, {0, 1}, {1, 0}, {-1, 0}};
+        char[] commands = {'^', '>', '>', '<'};
+        for (int i = 0; i < 4; i++) {
+            queue.add(new Move(start + directions[i][0], start + directions[i][1], commands[i]));
+        }
+
+        queue.add(new Move(start, start - 1, '^'));
+        queue.add(new Move(start, start + 1, '>'));
+        queue.add(new Move(start + 1, start, '>'));
+        queue.add(new Move(start - 1, start, '<'));
 
         while (!queue.isEmpty()) {
             Move move = queue.poll();
-            if (move.x < 0 || move.x >= width || move.y < 0 || move.y >= height) continue;
+            if (move.x < 0 || move.x >= size || move.y < 0 || move.y >= size || visited[move.x][move.y]) continue;
+            visited[move.x][move.y] = true;
             if (obstacles.contains("" + grid[move.x][move.y])) continue;
             if (grid[move.x][move.y] == '@') return move.direction;
 
-            queue.add(new Move(move.x, move.y - 1, move.direction));
-            queue.add(new Move(move.x, move.y + 1, move.direction));
-            queue.add(new Move(move.x + 1, move.y, move.direction));
-            queue.add(new Move(move.x - 1, move.y, move.direction));
+            for (int[] direction : directions) {
+                queue.add(new Move(move.x + direction[0], move.y + direction[1], move.direction));
+            }
         }
 
         System.err.println("No path found");