[C++] [Simulation] 백준 14503번 : 로봇 청소기

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

14503번: 로봇 청소기

#include<iostream>
#include<vector>
#include<utility>

using namespace std;

vector<vector<int>> map;
int direct[4][2] = { {-1,0},{0,1},{1,0},{0,-1} };
int result = 0;
int N, M;		//맵 크기
int r, c, d;	//로봇좌표(r, c) , 바라보는 방향 d (0=북, 1=동, 2=남, 3=서)

void solution() {
	pair<pair<int, int>, int> now = make_pair(make_pair(r, c), d);

	while (1) {
		if(map[now.first.first][now.first.second] == 0){
			map[now.first.first][now.first.second] = 2;		//청소한 곳은 2 로 표시
			result++;
		}

		pair<pair<int, int>, int> temp = now;

		//각 방향 탐색
		for (int i = 0; i < 4; i++) {
			now.second = (now.second + 3) % 4;	//현재방향 기준 왼쪽으로 회전
			int y = now.first.first + direct[now.second][0];
			int x = now.first.second + direct[now.second][1];

			if (0 <= y && y < N && 0 <= x && x < M) {
				if (map[y][x] == 0) {
					now = make_pair(make_pair(y, x), now.second);
					break;
				}
			}
		}

		//네 방향 모두 청소 되었거나 벽이라면
		if (now == temp) {
			int y = now.first.first + direct[(now.second + 2) % 4][0];			//뒤로이동
			int x = now.first.second + direct[(now.second + 2) % 4][1];	//뒤로이동

			if(0 <= y && y < N && 0 <= x && x < M) {
				if (map[y][x] == 1)
					return;
				else {
					now = make_pair(make_pair(y, x), now.second);
				}
			}
		}
	}
}

int main(void) {
	cin >> N >> M;
	cin >> r >> c >> d;

	//맵 입력
	for (int i = 0; i < N; i++) {
		vector<int> v;
		for (int j = 0; j < M; j++) {
			int temp;
			cin >> temp;
			v.push_back(temp);
		}
		map.push_back(v);
	}

	solution();

	cout << result << endl;
}