이것이 취업을 위한 코딩 테스트다 with 파이썬을 공부하면서 정리한 내용입니다.


문제 내용

  • 캐릭터가 있는 장소는 1×11 \times 1 크기의 정사각형으로 이뤄진 N×MN \times M 크기의 직사각형
  • 각각의 칸은 육지 또는 바다
  • 캐릭터는 동서남북 중 한 곳을 바라봄
  • 맵의 각 칸은 (A,B)(A, B)로 나타낼 수 있고 A는 북쪽으로부터 떨어진 칸의 개수, B는 서쪽으로부터 떨어진 칸의 개수
  • 캐릭터는 상하좌우로 움직일 수 있고, 바다로 되어 있는 공간에는 갈 수 없음
  • 캐릭터의 움직임을 설정하기 위해 정해 놓은 매뉴얼은 다음과 같음
    1. 현재 위치에서 현재 방향을 기준으로 왼쪽 방향부터 차례대로 갈 곳을 정함
    2. 캐릭터의 바로 왼쪽 방향에 아직 가보지 않은 칸이 존재한다면, 왼쪽 방향으로 회전한 다음 왼쪽으로 한 칸 전진. 왼쪽 방향에 가보지 않은 칸이 없다면, 왼쪽 방향으로 회전만 수행하고 1단계로 돌아감
    3. 만약 네 방향 모두 이미 가본 칸이거나 바다로 되어 있는 칸인 경우에는, 바라보지 않은 방향을 유지한 채로 한 칸 뒤로 가고 1단계로 돌아감. 단, 이때 뒤쪽 방향이 바다인 칸이라 뒤로 갈 수 없는 경우에는 움직임을 멈춤
  • 매뉴얼에 따라 캐릭터를 이동시킨 뒤에, 캐릭터가 방문한 칸의 수를 출력하는 프로그램 작성

입력 조건

  • 첫째 줄에 세로 크기 N과 가로 크기 M을 공백으로 구분하여 입력
  • 둘째 줄에 게임 캐릭터가 있는 칸의 좌표 (A,B)(A, B)와 바라보는 방향 d가 각각 서로 공백으로 구분하여 주어짐, 방향 d의 값으로는 다음과 같이 4가지가 존재
    • 0: 북쪽
    • 1: 동쪽
    • 2: 남쪽
    • 3: 서쪽
  • 셋째 줄부터 맵이 육지인지 바다인지에 대한 정보가 주어짐. N개의 줄에 맵의 상태가 북쪽부터 남쪽 순서대로, 각 줄의 데이터는 서쪽부터 동쪽 순서대로 주어짐. 맵의 외곽은 항상 바다로 되어 있음. 처음에 게임 캐릭터가 위치한 칸의 상태는 항상 육지
    • 0: 육지
    • 1: 바다

출력 조건

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

입력 예시

4 4
1 1 0
1 1 1 1
1 0 0 1
1 1 0 1
1 1 1 1

출력 예시

3

문제 해설

  • 전형적인 시뮬레이션 문제
  • 별도의 알고리즘이 필요하기보다는 문제에서 요구하는 내용을 오류 없이 구현해야 함
  • 일반적으로 방향을 설정해서 이동하는 문제 유형에서는 dx, dy라는 별도의 리스트를 만들어 방향을 정하는 것이 효과적
    • 예를 들어, 캐릭터가 북쪽을 바라보고 있을 때 북쪽으로 이동하기 위해 xy 좌표에 각각 dx[0], dy[0]만큼 더함. 즉, 현재 위치에서 (1,0)(-1, 0)만큼 이동하는 것
  • 반복문을 이용하여 모든 방향을 차례대로 확인할 수 있음
  • 파이썬에서 2차원 리스트를 선언할 때는 컴프리헨션을 이용하는 것이 효율적
  • 함수 바깥에서 선언된 전역 변수가 함수 안에서 사용될 때는 global 키워드 사용

소스 코드

# N, M을 공백으로 구분하여 입력받기
n, m = map(int, input().split())

# 방문한 위치를 저장하기 위한 맵을 생성하여 0으로 초기화
d = [[0] * m for _ in range(n)]
# 현재 캐릭터의 X 좌표, Y 좌표, 방향을 입력받기
x, y, direction = map(int, input().split())
d[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 direction
    direction -= 1
    if direction == -1:
        direction = 3

# 시뮬레이션 시작
count = 1
turn_time = 0
while True:
    # 왼쪽으로 회전
    turn_left()
    nx = x + dx[direction]
    ny = y + dy[direction]
    # 회전한 이후 정면에 가보지 않은 칸이 존재하는 경우 이동
    if d[nx][ny] == 0 and array[nx][ny] == 0:
        d[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 array[nx][ny] == 0:
            x = nx
            y = ny
        # 뒤가 바다로 막혀있는 경우
        else:
            break
        turn_time = 0

# 정답 출력
print(count)

나의 풀이

  • 입력으로 주어지는 맵과 캐릭터가 움직이면서 알아내는 맵을 선언
  • 캐릭터가 이동할 수 있는 움직임을 정의

소스 코드

import sys

n, m = map(int, sys.stdin.readline().split())
a, b, d = map(int, sys.stdin.readline().split())
world_map = [list(map(int, sys.stdin.readline().split())) for _ in range(n)]
player_map = [[-1] * m for _ in range(n)]

# 방향별 움직임
steps = ((-1, 0), (0, 1), (1, 0), (0, -1))

player_map[a][b] = 0


count = 1
turn_time = 0
while True:
    # 현재 방향을 기준으로 왼쪽 방향으로 회전
    d = d - 1 if d > 0 else 3
    na = a + steps[d][0]
    nb = b + steps[d][1]
    # 아직 가보지 않은 칸이면
    if player_map[na][nb] == -1:
        # 왼쪽으로 회전한 횟수 초기화
        turn_time = 0
        # 이동하려는 곳이 육지면
        if world_map[na][nb] == 0:
            # 캐릭터가 가진 지도에 해당 위치를 육지로 표시
            player_map[na][nb] = 0
            # 캐릭터를 해당 위치로 이동
            a = na
            b = nb
            count += 1
        # 이동하려는 곳이 바다면
        else:
            # 캐릭터가 가진 지도에 해당 위치를 바다로 표시
            player_map[na][nb] = 1
    # 가본 칸이면
    else:
        # 왼쪽으로 회전한 횟수 + 1
        turn_time += 1
        # 네 방향 모두 가보지 않았으면 다시 1단계로
        if turn_time < 4:
            continue
        # 네 방향 모두 가봤다면
        else:
            na = a - steps[d][0]
            nb = b - steps[d][1]
            # 가본 방향의 수 초기화
            turn_time = 0
            # 이동하려는 곳이 육지면
            if world_map[na][nb] == 0:
                # 캐릭터를 해당 위치로 이동
                a = na
                b = nb
            else:
                break

print(count)
profile
전부인 것처럼, 전부가 아닌 것처럼

0개의 댓글