[Programmers] 미로 탈출 (JAVA)

문제 설명

1 x 1 크기의 칸들로 이루어진 직사각형 격자 형태의 미로에서 탈출하려고 합니다. 각 칸은 통로 또는 벽으로 구성되어 있으며, 벽으로 된 칸은 지나갈 수 없고 통로로 된 칸으로만 이동할 수 있습니다. 통로들 중 한 칸에는 미로를 빠져나가는 문이 있는데, 이 문은 레버를 당겨서만 열 수 있습니다. 레버 또한 통로들 중 한 칸에 있습니다. 따라서, 출발 지점에서 먼저 레버가 있는 칸으로 이동하여 레버를 당긴 후 미로를 빠져나가는 문이 있는 칸으로 이동하면 됩니다. 이때 아직 레버를 당기지 않았더라도 출구가 있는 칸을 지나갈 수 있습니다. 미로에서 한 칸을 이동하는데 1초가 걸린다고 할 때, 최대한 빠르게 미로를 빠져나가는데 걸리는 시간을 구하려 합니다.

미로를 나타낸 문자열 배열 maps가 매개변수로 주어질 때, 미로를 탈출하는데 필요한 최소 시간을 return 하는 solution 함수를 완성해주세요. 만약, 탈출할 수 없다면 -1을 return 해주세요.

---

제한사항

• 5 ≤ maps의 길이 ≤ 100
  • 5 ≤ maps[i]의 길이 ≤ 100
  • maps[i]는 다음 5개의 문자들로만 이루어져 있습니다.
    • S : 시작 지점
    • E : 출구
    • L : 레버
    • O : 통로
    • X : 벽
  • 시작 지점과 출구, 레버는 항상 다른 곳에 존재하며 한 개씩만 존재합니다.
  • 출구는 레버가 당겨지지 않아도 지나갈 수 있으며, 모든 통로, 출구, 레버, 시작점은 여러 번 지나갈 수 있습니다.

 

import java.util.LinkedList;
import java.util.Queue;
import java.util.stream.IntStream;

public class 미로탈출 {
  public int solution(String[] maps) {
    char[][] graph = new char[maps.length][maps[0].length()];
    Position startPosition = null;
    Position leverPosition = null;
    Position exitPosition = null;
    for (int x : IntStream.range(0, maps.length).toArray()) {
      for (int y : IntStream.range(0, maps[x].length()).toArray()) {
        char currentChar = maps[x].charAt(y);
        graph[x][y] = currentChar;
        if (currentChar == 'S') {
          startPosition = new Position(x, y, 0);
        } else if (currentChar == 'L') {
          leverPosition = new Position(x, y, 0);
        } else if (currentChar == 'E') {
          exitPosition = new Position(x, y, 0);
        }
      }
    }

    int startPointToLeverPoint = bfs(graph, startPosition, leverPosition);
    int leverPointToExitPoint = bfs(graph, leverPosition, exitPosition);
    return startPointToLeverPoint == -1 || leverPointToExitPoint == -1
            ? -1
            : startPointToLeverPoint + leverPointToExitPoint;
  }

  private int bfs(char[][] graph, Position startingPosition, Position destinationPosition) {
    int[] dX = { -1, 0, 1, 0 };
    int[] dY = { 0, 1, 0, -1 };
    int maxHeight = graph.length;
    int maxWidth = graph[0].length;
    boolean[][] visitedPoint = new boolean[maxHeight][maxWidth];
    Queue<Position> queue = new LinkedList<Position>();
    queue.add(new Position(startingPosition.x, startingPosition.y, 0));
    visitedPoint[startingPosition.x][startingPosition.y] = true;

    while(!queue.isEmpty()) {
      Position currentPosition = queue.poll();
      int x = currentPosition.x;
      int y = currentPosition.y;
      if (x == destinationPosition.x && y == destinationPosition.y) {
        return currentPosition.count;
      }

      for (int i : IntStream.range(0, dX.length).toArray()) {
        int currentX = x + dX[i];
        int currentY = y + dY[i];
        if(currentX < 0 || currentX >= maxHeight || currentY < 0 || currentY >= maxWidth) {
          continue;
        }
        if (!visitedPoint[currentX][currentY] && graph[currentX][currentY] != 'X') {
          queue.add(new Position(currentX, currentY, currentPosition.count + 1));
          visitedPoint[currentX][currentY] = true;
        }
      }
    }

    return -1;
  }

  static class Position {
    public int x;
    public int y;
    public int count;
    Position (int x, int y, int count) {
      this.x = x;
      this.y = y;
      this.count = count;
    }
  }
}