- 문제

게임 캐릭터가 맵 안에서 움직이는 시스템을 개발 중이다. 캐릭터가 있는 장소는 1 X 1 크기의 정사각형으로 이뤄진 N X M 크기의 직사각형으로, 각각의 칸은 육지 또는 바다이다. 캐릭터는 동서남북 중 한 곳을 바라본다.
맵의 각 칸은 (A, B)로 나타낼 수 있고, A는 북쪽으로부터 떨어진 칸의 개수, B는 서쪽으로부터 떨어진 칸의 개수이다. 캐릭터는 상하좌우로 움직일 수 있고, 바다로 되어 있는 공간에는 갈 수 없다. 캐릭터의 움직임을 설정하기 위해 정해 놓은 메뉴얼은 다음과 같다.

  1. 현재 위치에서 현재 방향을 기준으로 왼쪽 방향(반시계 방향으로 90도 회전한 방향)부터 차례대로 갈 곳을 정한다.
  2. 캐릭터의 바로 왼쪽 방향에 아직 가보지 않은 칸이 존재한다면, 왼쪽 방향으로 회전한 다음 왼쪽으로 한 칸 전진한다. 왼쪽 방향에 가보지 않은 칸이 없다면, 왼쪽 방향으로 회전만 수행하고 1단계로 돌아간다.
  3. 만약 네 방향 모두 이미 가본 칸이거나 바다로 되어 있는 칸인 경우에는, 바라보는 방향을 유지한 채로 한 칸 뒤로 가고 1단계로 돌아간다. 단, 이때 뒤쪽 방향이 바다인 칸이라 뒤로 갈 수 없는 경우에는 움직임을 멈춘다.

위 과정을 반복적으로 수행하면서 캐릭터의 움직임에 이상이 있는지 테스트하려고 한다. 메뉴얼에 따라 캐릭터를 이동시킨 뒤에, 캐릭터가 방문한 칸의 수를 출력하는 프로그램을 만드시오.

- 입력 조건

첫째 줄에 맵의 세로 크기 N과 가로 크기 M을 공백으로 구분하여 입력 (3 <= N, M <= 50)
둘째 줄에 게임 캐릭터가 있는 칸의 좌표 (A, B)와 바라보는 방향 d가 각각 서로 공백으로 구분하여 주어짐
방향 d의 값으로는 다음과 같이 4가지가 존재

0: 북쪽
1: 동쪽
2: 남쪽
3: 서쪽

셋째 줄부터 맵이 육지인지 바다인지에 대한 정보가 주어짐
N개의 줄에 맵의 상태가 북쪽부터 남쪽 순서대로, 각 줄의 데이터는 서쪽부터 동쪽 순서대로 주어짐
맵의 외곽은 항상 바다

0: 육지
1: 바다

처음에 게임 캐릭터가 위치한 칸의 상태는 항상 육지

- 출력 조건

첫째 줄에 이동을 마친 후 캐릭터가 방문한 칸의 수를 출력

- 입력 예시

4 4 # 4 X 4 맵 생성
1 1 0 # (1, 1)에 북쪽(0) 을 바라보고 서 있는 캐릭터
1 1 1 1 # 첫 줄은 모두 바다
1 0 0 1 # 둘째 줄은 바다/육지/육지/바다
1 1 0 1 # 셋째 줄은 바다/바다/육지/바다
1 1 1 1 # 넷째 줄은 모두 바다

- 출력 예시

3

- 문제 풀이

개인적으로, 이 문제는 DP 문제라고 생각하는게 더 좋지 않을까 생각한다. 다음 위치로 이동하기 위해서는, 그 위치가 육지이면서 이전에 방문한 적이 없어야한다. 따라서, 문제를 풀기 위해서는 이전에 방문한 적이 있는지를 확인할 수 있는 변수가 필요하다. 이런 방식의 풀이는 DP 문제에서 자주 사용되는 풀이라고 생각한다.
어쨌든, 현재 위치에서 반시계 방향으로 90도 회전한 뒤, 앞에 있는 위치가 갈 수 있는 위치인지 (바다인지 육지인지, 이전에 방문한 적이 있는지 없는지)를 확인해야한다. 따라서 방향을 정의하는 dx와 dy의 순서를 북, 동, 남, 서 순으로 정의했다. (처음 시작 방향이 북쪽이고, 이후 반시계 방향으로 90도씩 회전하면 서쪽 -> 남쪽 -> 동쪽 순으로 회전함)
다음으로, 모든 항목이 0으로 이루어진, 지도와 같은 크기의 배열을 생성한다. 이는 이전에 해당 지역을 방문한 적이 있는지 없는지 확인하는 변수가 된다.

프로세스를 순차적으로 나열해보면,
1. 방문 기록을 저장하는 2차원 배열 생성
2. 시작 지점에서 왼쪽으로 회전하면서 앞에 있는 위치가 바다인지 육지인지, 이전에 방문한 경험이 있는지 확인
3. 육지이면서 방문한 경험이 없다면 해당 위치로 이동 후 count 증가
4. 바다이거나 방문한 경험이 있다면 이동하지않고 다시 반시계 방향으로 90도 회전
5. 네 방향 모두 이동할 수 있는 위치가 없다면 방향을 바꾸지 않고 뒤로 이동
6. 이동할 수 있는 위치가 없다면 count를 출력

n, m = map(int, input().split())
x, y, d = map(int, input().split())

visited = [[0] * m for _ in range(n)]
visited[x][y] = 1

array = []
for i in range(n):
    array.append(list(map(int, input().split())))
    
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]

def turn_left():
    global d
    d -= 1
    if d == -1:
        d = 3
        
count = 1
turn_time = 0

while True:
    turn_left()
    nx = x + dx[d]
    ny = y + dy[d]
    
    if visited[nx][ny] == 0 and array[nx][ny] == 0:
        visited[nx][ny] = 1
        x = nx
        y = ny
        count += 1
        turn_time = 0
        continue
    else:
        turn_time += 1
        
    if turn_time == 4:
        nx = x - dx[d]
        ny = y - dy[d]
        
        if array[nx][ny] == 0:
            x = nx
            y = ny
        else:
            break
        turn_time = 0
        
print(count)

출처: 이것이 취업을 위한 코딩 테스트다 with 파이썬, 나동빈 지음

profile
코딩하는 물리학도

0개의 댓글