이것이 코딩테스트다 - 5일차

주제

완전탐색

문제

현민이는 게임 캐릭터가 맵 안에서 움직이는 시스템을 개발 중이다. 캐릭터가 있는 장소는 1 x 1 크기의 정사각형으로 이루어진 N x M 크기의 직사각형으로 각각의 칸은 육지 또는 바다이다. 캐릭터는 동서남북 중 한 곳을 바라본다.

맵의 각 같은 (A, B)로 나타낼 수 있고, A는 북쪽으로부터 떨어진 칸의 개수, B는 서쪽으로부터 떨어진 칸의 개수이다. 캐릭터는 상하좌우로 움직일 수 있고, 바다로 되어 있는 공간에는 갈 수 없다. 캐릭터의 움직임을 설정하기 위해 정해 놓은 매뉴얼은 이러하다.

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

현민이는 위 과정을 반복적으로 수행하면서 캐릭터의 움직임에 이상이 있는지 테스트하려고 한다.

매뉴얼에 따라 캐틱터를 이동 시킨 후, 캐릭터가 방문한 칸의 수를 출력하는 프로그램을 만드시오.

예시

# 바다 = 1, 캐릭터가 움직임 = 2 프로그래밍에서는 전부 1로 표현할 예정

# 첫번째 반복
'''
1 1 1 1
1 2 0 1
1 1 0 1
1 1 1 1
'''

# 두번째 반복
'''
1 1 1 1
1 2 2 1
1 1 0 1
1 1 1 1
'''

# 세번째 반복
'''
1 1 1 1
1 2 2 1
1 1 2 1
1 1 1 1
'''

조건

첫째 줄에 맵의 세로 크기 N과 가로 크기 M을 공백으로 구분하여 입력한다. (3 <= N, M <= 50)

둘째 줄에 게임 캐릭터가 있는 칸의 좌표 (A,B)와 바라보는 방향 d가 각각 서로 공백으로 구분하여 주어진다. 방향 d의 값으로는 다음과 같이 4가지가 존재한다.

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

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

• 0 : 육지
• 1 : 바다

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

문제 풀이 아이디어

북(위), 동(오른쪽), 남(아래), 서(왼쪽) 라는 조건을 통해 아래와 같이 dx, dy를 설정한다.

• dx = [-1, 0, 1, 0]
• dy = [0, 1, 0, -1]

문제 풀이 방법

1. N, M, x, y, direction을 입력받고 game_map을 초기화한다.
2. 현재 캐릭터가 서있는 자리를 1로 초기화 하고 dx, dy를 초기화한다.
3. 캐릭터가 왼쪽을 볼 수 있도록 하는 함수를 만든다.
4. 무한 루프를 돌면서 왼쪽 방향을 보는 함수를 실행하고, nx, ny에 현재 위치와 방향을 반복해서 더한다.
5. 캐릭터가 보고 있는 방향에서 게임맵이 0(육지)이라면 그 위치를 갔음을 표시(=1)하고 x, y 위치를 nx, ny의 위치로 갱신한다. 또한 캐릭터 방문 칸 수를 세고, 4번으로 다시 돌아간다.
6. 만약 캐릭터가 보고 있는 방향이 육지가 아니라면 캐릭터가 방향을 돌렸던 것을 체크한다.
7. 캐릭터가 방향을 돌린 것이 4번이면 뒤로 돌아가는 로직을 작성하고 뒤로 돌아간 곳이 0이라면 다시 4번부터 시작할 수 있도록 nx, ny의 위치를 갱신한다.
8. 만약 뒤로 돌아간 곳이 1(=바다 또는 갔던 곳) 루프에서 빠져나온다.

N, M = map(int, input().split())
x, y, direction = map(int, input().split())


game_map = []
for _ in range(N):
    game_map.append(list(map(int, input().split())))

# 현재 캐릭터가 서있는 자리
game_map[x][y] = 1

dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]

# 캐릭터가 왼쪽방향으로 보는 함수
def left_direction():
    global direction
    direction -= 1
    if direction == -1:
        direction = 3 

count = 1
turn_time = 0

while True:
    left_direction()
    nx = x + dx[direction]
    ny = y + dy[direction]

    if game_map[nx][ny] == 0:
        game_map[nx][ny] = 1
        x = nx
        y = ny
        count += 1
        turn_time = 0
        continue
    else:
        turn_time += 1
    
    if turn_time == 4:
      nx = x - dx[direction]
      ny = y - dy[direction]

      if game_map[nx][ny] == 0:
          x = nx
          y = ny
      else:
          break
      turn_time = 0
    
print(count)