[BOJ] 14503 - 로봇 청소기

개인적인 풀이일 뿐, 최적의 정답이 아님을 알려드립니다.

 

문제

www.acmicpc.net/problem/14503

14503번: 로봇 청소기

난이도: 골드 5

사용언어: JAVA

 

풀이

사실 어떤 알고리즘 문제가 아니라 구현 문제라서, 문제의 조건대로 쭉 구현해주시면 됩니다. 그리고 주의해야 할 점은 청소는 북-서-남-동으로 이루어지지만, 입력을 받을 때 d 값은 0이 북쪽, 1이 동쪽, 2가 남쪽, 3이 서쪽으로 방향이 반대라는 점입니다. 

 

저는 3차원 배열을 이용해 현재의 방향과 이동할 위치를 구현했습니다.

 

import java.io.*;
import java.util.*;
 
class Main {
 
    static int N;
    static int M;
    static int[][] arr;
    static int cnt = 0;
    static int[][][] dir = {{{0, -1}, {1, 0}, {0, 1}, {-1, 0}},
            {{1, 0}, {0, 1}, {-1, 0}, {0, -1}},
            {{0, 1}, {-1, 0}, {0, -1}, {1, 0}},
            {{-1, 0}, {0, -1}, {1, 0}, {0, 1}}}; // 청소기의 방향에 따른 회전 위치
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
 
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        st = new StringTokenizer(br.readLine());
        int r, c, d;
        int i;
        arr = new int[N][M];
        r = Integer.parseInt(st.nextToken());
        c = Integer.parseInt(st.nextToken());
        d = Integer.parseInt(st.nextToken());
 
        if (d == 1) // d의 값을 배열의 순서에 맞게 조정
            d = 3;
        else if (d == 3)
            d = 1;
 
        for (i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < M; j++) {
                arr[i][j] = Integer.parseInt(st.nextToken());
            }
        }
        cnt++;
        arr[r][c] = 2;
        while (true) {
            for (i = 0; i < 4; i++) {
                if (arr[r + dir[d][i][0]][c + dir[d][i][1]] == 0) { 왼쪽부터 청소를 했는지 확인
                    cnt++;
                    r = r + dir[d][i][0]; // 청소가 안되었다면, 이동
                    c = c + dir[d][i][1];
                    arr[r][c] = 2; // 청소 했다고 표시
                    d = (d + i + 1) % 4; // 방향도 업데이트
                    break;
                }
            }
            if (i == 4) { // 만일 주위에 아직 청소된 칸이 없다면
                r = r + dir[d][1][0]; // 뒤로 한 칸 후진
                c = c + dir[d][1][1];
                if (arr[r][c] == 1) { // 뒤가 벽이라면
                    bw.write(String.valueOf(cnt)); // 종료
                    bw.flush();
                    bw.close();
                    br.close();
                    System.exit(0);
                }
            }
        }
    }
}