[백준][14503] 로봇 청소기

이 포스트는 백준 사이트의 로봇 청소기 문제 풀이입니다.

문제

코드 구현

문제에 주어진 로봇 청소기의 작업 순서를 코드에 그대로 표현하고자 하였습니다.

먼저, 현재 위치에서 is_dirty_tiles_left() 함수를 실행하여 지금 해야하는 작업을 실행합니다.
그리고 변경된 위치를 작업 큐에 넣는 것을 반복하여 큐가 빌 때 까지 반복합니다.

마지막으로 청소된 타일의 개수를 출력하였습니다.

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

#define IO_SETTING std::ios::sync_with_stdio(false)
#define INPUT_SETTING std::cin.tie(0)

enum TILE_TYPE {
	DIRTY = 0,
	WALL, 
	CLEAN
};

int N, M;
int robot_row, robot_col, robot_dir;
vector<vector<TILE_TYPE>> board;
int dir_row[4]{-1, 0, 1, 0};
int dir_col[4]{0, 1, 0, -1};

void input() {
	cin >> N >> M;
	cin >> robot_row >> robot_col >> robot_dir;
	board.assign(N, vector<TILE_TYPE>(M, TILE_TYPE::DIRTY));
	for(int row = 0; row < N; ++row) {
		for(int col = 0, temp; col < M; ++col) {
			cin >> temp;
			board[row][col] = TILE_TYPE(temp);
		}
	}
}

bool is_dirty_tiles_left(int& curr_row, int& curr_col) {
	if(TILE_TYPE::DIRTY == board[curr_row][curr_col]) {
		board[curr_row][curr_col] = TILE_TYPE::CLEAN;
		return true;
	} 

	int dirty_tile_count = 0;
	for(int dir = 0; dir < 4; ++dir) {
		int new_row = curr_row + dir_row[dir];
		int new_col = curr_col + dir_col[dir];

		if(new_row < 0 || new_row >= N || new_col < 0 || new_col >= M)
			continue;
		if(TILE_TYPE::DIRTY == board[new_row][new_col]) {
			dirty_tile_count += 1;
		}
	}

	if(dirty_tile_count == 0) {
		int robot_back_dir = (robot_dir + 2) % 4;
		int new_row = curr_row + dir_row[robot_back_dir];
		int new_col = curr_col + dir_col[robot_back_dir];

		if(TILE_TYPE::WALL != board[new_row][new_col]) {
			curr_row = new_row;
			curr_col = new_col;
		} else {
			return false;
		}
	} else {
		for(int turn = 0; turn < 4; ++turn) {
			robot_dir = (robot_dir + 3) % 4;
			int new_row = curr_row + dir_row[robot_dir];
			int new_col = curr_col + dir_col[robot_dir];

			if(TILE_TYPE::DIRTY == board[new_row][new_col]) {
				curr_row = new_row;
				curr_col = new_col;
				return true;
			}
		}
	}
	return true;
}

void cleanup_tile() {
	queue<pair<int, int>> q;
	q.push({robot_row, robot_col});

	while(false == q.empty()) {
		int curr_row = q.front().first;
		int curr_col = q.front().second;
		q.pop();

		if(is_dirty_tiles_left(curr_row, curr_col)) {
			q.push({curr_row, curr_col});
		}
	}
}

void print_cleanup_tile_count() {
	int count = 0;
	for(int row = 0; row < N; ++row) {
		for(int col = 0; col < M; ++col) {
			if(board[row][col] == TILE_TYPE::CLEAN) {
				count += 1;
			}
		}
	}
	printf("%d", count);
}

int main() {
	input();
	cleanup_tile();
	print_cleanup_tile_count();
	return 0;
} 

결과