백준 14503
로봇 청소기가 주어졌을 때, 청소하는 영역의 개수를 구하기.
- 로봇 청소기가 있는 장소는 N×M 크기의 직사각형으로 나타낼 수 있으며, 1×1크기의 정사각형 칸으로 나누어져 있다. 각각의 칸은 벽 또는 빈 칸이다. 청소기는 바라보는 방향이 있으며, 이 방향은 동, 서, 남, 북중 하나이다. 지도의 각 칸은 (r, c)로 나타낼 수 있고, r은 북쪽으로부터 떨어진 칸의 개수, c는 서쪽으로 부터 떨어진 칸의 개수이다.
- 로봇 청소기는 다음과 같이 작동한다.
- 현재 위치를 청소한다.
- 현재 위치에서 현재 방향을 기준으로 왼쪽 방향부터 차례대로 인접한 칸을 탐색한다.
a. 왼쪽 방향에 아직 청소하지 않은 공간이 존재한다면, 그 방향으로 회전한 다음 한 칸을 전진하고 1번부터 진행한다.
b. 왼쪽 방향에 청소할 공간이 없다면, 그 방향으로 회전하고 2번으로 돌아간다.
c. 네 방향 모두 청소가 이미 되어있거나 벽인 경우에는, 바라보는 방향을 유지한 채로 한 칸 후진을 하고 2번으로 돌아간다.
d. 네 방향 모두 청소가 이미 되어있거나 벽이면서, 뒤쪽 방향이 벽이라 후진도 할 수 없는 경우에는 작동을 멈춘다.
로봇 청소기는 이미 청소되어있는 칸을 또 청소하지 않으며, 벽을 통과할 수 없다.
- 첫째 줄에 세로 크기 N과 가로 크기 M이 주어진다. (3 ≤ N, M ≤ 50)
- 둘째 줄에 로봇 청소기가 있는 칸의 좌표 (r, c)와 바라보는 방향 d가 주어진다. d가 0인 경우에는 북쪽을, 1인 경우에는 동쪽을, 2인 경우에는 남쪽을, 3인 경우에는 서쪽을 바라보고 있는 것이다.
- 셋째 줄부터 N개의 줄에 장소의 상태가 북쪽부터 남쪽 순서대로, 각 줄은 서쪽부터 동쪽 순서대로 주어진다. 빈 칸은 0, 벽은 1로 주어진다. 지도의 첫 행, 마지막 행, 첫 열, 마지막 열에 있는 모든 칸은 벽이다.
- 로봇 청소기가 있는 칸의 상태는 항상 빈 칸이다.
입출력
| 입력 | 출력 |
|---|---|
| 3 3 1 1 0 1 1 1 1 0 1 1 1 1 | 1 |
| 11 10 7 4 0 1 1 1 1 1 1 1 1 1 1 1 0 0 0 0 0 0 0 0 1 1 0 0 0 1 1 1 1 0 1 1 0 0 1 1 0 0 0 0 1 1 0 1 1 0 0 0 0 0 1 1 0 0 0 0 0 0 0 0 1 1 0 0 0 0 0 0 1 0 1 1 0 0 0 0 0 1 1 0 1 1 0 0 0 0 0 1 1 0 1 1 0 0 0 0 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 | 57 |
접근 방식
:
알게된 점
시뮬레이션 문제는 문제에서 나와있는대로 충실히 구현하면 된다. 특히 이 문제는 한 번에 움직일 수 있는 한 방향이 정해져 있으므로 특별한 알고리즘도 필요하지 않다.
코드
n, m = list(map(int, input().split())) # n*m
r, c, d = list(map(int, input().split()))
# d가 0:북, 1:동, 2:남, 3:서
maps = [list(map(int, input().split())) for _ in range(n)] # 빈칸0, 벽1
# 방향 바꾸기 함수
def change_direction(d):
if d==0: return 3
elif d==1: return 0
elif d==2: return 1
elif d==3: return 2
# (있는 자리)청소하기 함수
def clean(x, y, d):
maps[x][y] = -1
cnt = 1
# 북동남서
dx = [-1,0,1,0]
dy = [0,1,0,-1]
while True:
flag = 0
# 한 자리에서 네번 돌 수 있음.
for _ in range(4):
nd = change_direction(d) #방향바꾸기
nx = x + dx[nd]
ny = y + dy[nd]
d = nd
if 0<=nx<n and 0<=ny<m: #map안에 있어야함
if maps[nx][ny] == 0:
maps[nx][ny] = -1 #청소
x, y = nx, ny #이동
flag += 1 #청소 여부
cnt += 1 #청
break
if flag == 0 : #본인 제외 청소 하나도 못함 or 모두 청소
nx = x - dx[d] #후진
ny = y - dy[d]
if 0<=nx<n and 0<=ny<m:
if maps[nx][ny] != 1: #map안에 있으면서 벽만 아니면 후진.
x, y = nx, ny
else:
return cnt
else:
return cnt
print(clean(r, c, d))
