[BOJ][Java] 14503. 로봇 청소기

문제

https://www.acmicpc.net/problem/14503

 

문제 이해

N×M 크기의 방이 주어진다.
로봇 청소기는 아래 규칙에 따라 작동한다.

1. 현재 위치를 청소한다.
2. 현재 위치에서 왼쪽 방향부터 차례대로 탐색한다.
   • 아직 청소하지 않은 빈 칸이 있으면 그 칸으로 이동 후 1번으로 돌아간다.
   • 없으면 바라보는 방향을 유지한 채로 한 칸 후진할 수 있으면 후진, 아니면 작동을 멈춘다.

최종적으로 로봇 청소기가 청소하는 칸의 개수를 출력한다.

 

문제 풀이

시뮬레이션 구현

• 청소할 때마다 카운트를 증가
• 네 방향 중 청소되지 않은 칸이 있으면 회전 후 이동
• 네 방향 모두 불가능하면 후진, 후진도 불가능하면 종료

 

코드

import java.util.*;
import java.io.*;

public class Main {
	static int n, m, r, c, dir, ans = 0;
	static int[][] arr;

	// 상 우 하 좌
	static int[] dr = { -1, 0, 1, 0 };
	static int[] dc = { 0, 1, 0, -1 };

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		n = Integer.parseInt(st.nextToken());
		m = Integer.parseInt(st.nextToken());

		st = new StringTokenizer(br.readLine());
		r = Integer.parseInt(st.nextToken());
		c = Integer.parseInt(st.nextToken());
		dir = Integer.parseInt(st.nextToken());

		arr = new int[n][m];
		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(br.readLine());
			for (int j = 0; j < m; j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
			}
		}

		while (true) {
			first();

			if (arr[r + 1][c] == 0 || arr[r][c + 1] == 0 || arr[r - 1][c] == 0 || arr[r][c - 1] == 0) {
				third();
			} else {
				if (second())
					continue;
				else
					break;
			}
		}

		System.out.println(ans);
	}

	static void first() {
		if (arr[r][c] == 0) {
			arr[r][c] = 2;
			ans++;
		}
	}

	static boolean second() {
		int nr = r - dr[dir];
		int nc = c - dc[dir];

		if (arr[nr][nc] == 1)
			return false;

		r = nr;
		c = nc;
		return true;
	}

	static void third() {
		dir = (dir - 1) < 0 ? 3 : (dir - 1);

		int nr = r + dr[dir];
		int nc = c + dc[dir];

		if (arr[nr][nc] != 0)
			return;

		r = nr;
		c = nc;
	}
}