현민이는 게임 캐릭터가 맵 안에서 움직이는 시스템을 개발 중이다. 캐릭터가 있는 장소는 1 X 1 크기의 정사각형으로 이뤄진 N X M 크기의 직사각형으로, 각각의 칸은 육지 또는 바다이다. 캐릭터는 동서남북 중 한 곳을 바라본다.
맵의 각 칸은 (A, B)로 나타낼 수 있고, A는 북쪽으로부터 떨어진 칸의 개수, B는 서쪽으로부터 떨어진 칸의 개수이다. 캐릭터는 상하좌우로 움직일 수 있고, 바다로 되어 있는 공간에는 갈 수 없다. 캐릭터의 움직임을 설정하기 위해 정해 놓은 매뉴얼은 이러하다.
현재 위치에서 현재 방향을 기준으로 왼쪽 방향(반시계 방향으로 90도 회전한 방향)부터 차례대로 갈 곳을 정한다.
캐릭터의 바로 왼쪽 방향에 아직 가보지 않은 칸이 존재한다면, 왼쪽 방향으로 횐전한 다음 왼쪽으로 한 칸을 전진한다. 왼쪽 방향에 가보지 않은 칸이 없다면, 왼쪽 방향으로 회전만 수행하고 1단계로 돌아간다.
만약 네 방향 모두 이미 가본 칸이거나 바다로 되어 있는 칸인 경우에는, 바라보는 방향을 유지한 채로 한 칸 뒤로 가고 1단계로 돌아간다. 단, 이때 뒤쪽 방향이 바다인 칸이라 뒤로 갈 수 없는 경우에는 움직임을 멈춘다.
현민이는 위 과정을 반복적으로 수행하면서 캐릭터의 움직임에 이상이 있는지 테스트하려고 한다. 메뉴얼에 따라 캐릭터를 이동시킨 뒤에, 캐릭터가 방문한 칸의 수를 출력하는 프로그램을 만드시오.
첫째 줄에 맵의 세로 크기 N과 가로 크기 M을 공백으로 구분하여 입력한다.
(3 <= N, M <= 50)
둘째 줄에 게임 캐릭터가 있는 칸의 좌표 (A, B)와 바라보는 방햔 d가 각각 서로 공백으로 구분하여 주어진다. 방향 d의 값으로는 다음과 같이 4가지가 존재한다.
0 : 북쪽
1 : 동쪽
2 : 남쪽
3 : 서쪽
셋째 줄부터 맵이 육지인지 바다인지에 대한 정보가 주어진다. N개의 줄에 맵의 상태가 북쪽부터 남쪽 순서대로, 각 줄의 데이터는 서쪽부터 동쪽 순서대로 주어진다. 맵의 외각은 항상 바다로 되어 있다.
0 : 육지
1 : 바다
처음에 게임 캐릭터가 위치한 칸의 상태는 항상 육지이다.
첫째 줄에 이동을 마친 후 캐릭터가 방문한 칸의 수를 출력한다.
입력 예시
4 4
1 1 0 // (1, 1)에 북쪽(0)을 바라보고 서 있는 캐릭터
1 1 1 1
1 0 0 1
1 1 0 1
1 1 1 1
출력 예시
3
n, m = map(int, input().split())
x,y,side = map(int,input().split())
# 북, 서, 남, 동 0 3 2 1
dx = [-1,0,1,0]
dy = [0,-1,0,1]
check = [[False for _ in range(m)] for _ in range(n)]
arr = []
for _ in range(n):
arr.append(list(map(int, input().split())))
ans = 0
while True:
cnt = 0
for _ in range(4):
# 방향 틀어보기
side = (side + 3) % 4
cnt += 1
# 회전한 방향에 방문안했으면 방문, 아니면 또 회전
nx = x + dx[side]
ny = y + dy[side]
if nx < 0 or nx >= n or ny < 0 or ny >= m:
continue
if check[nx][ny] == True or arr[nx][ny] == 1:
continue
x = nx
y = ny
ans += 1
check[x][y] = True
break
# 네 방향 다 가보았을때, 육지인경우 방향유지하고 그대로 한칸뒤, 뒤가 바다면 종료
if cnt == 4:
x = x + dx[(side + 6) % 4]
y = y + dy[(side + 6) % 4]
if arr[x][y] == 1:
print(ans)
break
문제에서 0,1,2,3을 각각 북쪽, 동쪽, 남쪽, 서쪽으로 방향을 표현하였는데 방향은 왼쪽으로 회전하기 때문에 dx,dy 을 북,서,남,동 순으로 만들어 놓고 풀이를 하였습니다. 방향 회전 또한 왼쪽 회전을 (side + 3) % 4, 뒤로 가기는 (side + 6) % 4로 표현하여 네 방향을 모두 가보았을 때 그 뒤가 육지면 뒤로 가고, 바다이면 방문한 칸 수를 출력하도로 하였습니다.
처음에 문제를 봤을 때 바라보는 방향과 실제 회전하여 움직이는 방향이 달라 이해하는데 꽤 시간이 걸렸던 것 같습니다.
실제 풀이 시간 : 36분
이것이 코딩 테스트다 with 파이썬 - 나동빈 저