[JAVA]백준 14503. [G5]로봇 청소기

백준 14503. 로봇 청소기

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

14503번: 로봇 청소기

 

단순한 구현 문제였는데 문제 이해부터 풀이까지 시간이 걸렸던 문제...

조건을 잘 따라 구현하면 별다른 문제는 없는 거 같다.

반복문을 계속 돌면서 4방향 중에 청소를 할 곳이 있거나 없을 때의 로봇 청소기의 동작을 각 조건문 안에 작성을 하고 뒤로 후진해야할 때 그곳이 벽이라면 반복문을 탈출해서 청소한 곳을 카운트하는 식으로 구현

 

조건

전체 코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

class Robot {
	int x;
	int y;
	int d;

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

public class Main {
	static int N, M, count = 0;
	static int[][] map;
	static Robot robot;
	// 북동남서
	static int[] dx = { -1, 0, 1, 0 };
	static int[] dy = { 0, 1, 0, -1 };

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		N = Integer.parseInt(st.nextToken());
		M = Integer.parseInt(st.nextToken());
		map = new int[N][M];

		st = new StringTokenizer(br.readLine());
		robot = new Robot(Integer.parseInt(st.nextToken()), Integer.parseInt(st.nextToken()),
			Integer.parseInt(st.nextToken()));

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

		//청소시작
		clean();
		System.out.println(count);
	}

	static void clean() {
		// 시작 위치 청소 : -1
		map[robot.x][robot.y] = -1;

		while (true) {
			// 왼쪽 회전
			int dic = robot.d - 1;
			if (dic == -1) {
				dic = 3;
			}
			// 왼쪽 확인
			int px = robot.x + dx[dic];
			int py = robot.y + dy[dic];
			if (map[px][py] == 0) {
				// 왼쪽 청소해야되면 이동후 청소
				robot.x = px;
				robot.y = py;
				robot.d = dic;
				map[px][py] = -1;
			} else if (map[px][py] == 1 || map[px][py] == -1) {    // 왼쪽이 벽이거나 청소해야될 경우

				// 4방향 확인을 위한 check 변수
				boolean check = false;
				for (int i = 0; i < 4; i++) {
					int x = robot.x + dx[i];
					int y = robot.y + dy[i];

					if (map[x][y] != 0) {     // 4방향 벽이거나 청소해야될 경우 true
						check = true;
					} else {                  // 한방향이라도 청소해야될 경우 false
						check = false;
						break;
					}
				}

				// 4방향이 벽이거나 청소해야 될경우
				if (check) {
					// 방향에서 뒤로 한칸 이동할 때의 좌표
					int x = robot.x - dx[robot.d];
					int y = robot.y - dy[robot.d];

					// 뒤로 이동하는 곳이 벽일때 반복종료
					if (map[x][y] == 1) {
						break;
					} else {  // 뒤로 한칸 이동 가능할 때 이동
						robot.x = x;
						robot.y = y;
					}
				} else {    // 한방향이라도 청소가능할 때, 방향만 변경
					robot.d = dic;
				}

			}
		}

		// 청소한 곳 카운트 (-1)
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				if (map[i][j] == -1) {
					count++;
				}
			}
		}
	}
}