14503번 : 로봇 청소기 [Java]

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

14503번: 로봇 청소기

 

해결 방안

로봇 청소기의 작동 원리 순서대로 하나씩 체크하면서 청소를 할 수 있는지 확인한다.

 

한 개의 while문을 사용해 

1. 현재 자리가 청소 가능하다면 청소하고 바꿔주자

2. 왼쪽을 탐색해보고 청소 가능하다면 이동하자

3. 왼쪽이 청소 불가능하다면 다시 왼쪽을 보자

4. 한 바퀴를 돌며 다 확인했다면, 후진할 위치를 보자

5. 후진할 위치가 벽이면 멈추고 벽이 아니라면 이동하자

 

위 순서대로 작성해주면 된다.

 

전체 코드

import java.util.Scanner;

public class Main {
	static class Point {
		int x, y, d;

		Point(int x, int y, int d) {
			this.x = x;
			this.y = y;
			this.d = d;
		}
	}

	static int N, M;
	static int r, c, d;
	static int[][] map;
	static int[] dr = { -1, 0, 1, 0 };
	static int[] dc = { 0, 1, 0, -1 };
	static int cnt;
	static int ans;

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		N = sc.nextInt(); // 세로크기
		M = sc.nextInt(); // 가로크기
		r = sc.nextInt();
		c = sc.nextInt();
		d = sc.nextInt(); // 바라보는 방향 : 0북, 1동, 2남, 3서

		map = new int[N][M];

		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				map[i][j] = sc.nextInt();
			}
		}

		cnt = 0;
		ans = 0;

		search(r, c, d);
		System.out.println(ans);
	}

	private static void search(int r, int c, int d) {

		while (true) {

			if (map[r][c] == 0) {
				map[r][c] = -1;
				ans++;
			}

			int dd = (d + 3) % 4;
			int nx = r + dr[dd];
			int ny = c + dc[dd];
			if (map[nx][ny] == 0) { // 청소 가능
				r = nx;
				c = ny;
				d = dd;
				cnt = 0;
			} else { // 청소 불가능
				cnt++;
				d = dd;
			}

			if (cnt == 4) {
				int bx = r - dr[d];
				int by = c - dc[d];
				if (map[bx][by] == 1) {
					return;
				} else {
					r = bx;
					c = by;
					cnt = 0;
				}
			}
		}
	}
}