BOJ 1103 (파이썬)

🖋️ 문제

형택이는 1부터 9까지의 숫자와, 구멍이 있는 직사각형 보드에서 재밌는 게임을 한다.

일단 보드의 가장 왼쪽 위에 동전을 하나 올려놓는다. 그다음에 다음과 같이 동전을 움직인다.

동전이 있는 곳에 쓰여 있는 숫자 X를 본다.
위, 아래, 왼쪽, 오른쪽 방향 중에 한가지를 고른다.
동전을 위에서 고른 방향으로 X만큼 움직인다. 이때, 중간에 있는 구멍은 무시한다.
만약 동전이 구멍에 빠지거나, 보드의 바깥으로 나간다면 게임은 종료된다. 형택이는 이 재밌는 게임을 되도록이면 오래 하고 싶다.

보드의 상태가 주어졌을 때, 형택이가 최대 몇 번 동전을 움직일 수 있는지 구하는 프로그램을 작성하시오.

입력
줄에 보드의 세로 크기 N과 가로 크기 M이 주어진다. 이 값은 모두 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 보드의 상태가 주어진다. 쓰여 있는 숫자는 1부터 9까지의 자연수 또는 H이다. 가장 왼쪽 위칸은 H가 아니다. H는 구멍이다.

출력
첫째 줄에 문제의 정답을 출력한다. 만약 형택이가 동전을 무한번 움직일 수 있다면 -1을 출력한다.

🤔 접근

1. 4가지 방향으로 이동하면서 맵 밖으로 이동하거나 'H'을 만나면 함수 종료하고 이전 단계에서 다시 탐색 (DFS)
2. 이미 방문했던 곳에 다시 도착하면 print(-1); exit();
3. 이때, 시작점부터 임의의 한 지점까지 경로가 여러 개일 가능성이 있음(DP). 따라서 각 지점마다 동전을 최대로 움직인 횟수를 DP에 저장. 만약, DP[x][y] 보다 count + 1 이 작다면, 굳이 방문할 필요 없음.

💻 내 코드

import sys
sys.setrecursionlimit(10**7)

maxcount = 0
def dfs(x, y, count):
    global maxcount
    dp[x][y] = count
    visited[x][y] = True
    num = int(table[x][y])
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]
    for i in range(4):
        nx = x + (dx[i] * num)
        ny = y + (dy[i] * num)
        
        if (0<=nx<n) and (0<=ny<m):
            if visited[nx][ny]:
                print(-1)
                exit()
            if table[nx][ny] != 'H' and dp[nx][ny] < count+1:
                dp[nx][ny] = dp[x][y] + 1

                dfs(nx, ny, count+1)
                # 방문하고 돌아오면 다시 미방문 처리
                visited[nx][ny] = False
        
    maxcount = max(maxcount, count)
    return count


n, m = map(int, input().split())
table = [list(sys.stdin.readline().strip()) for _ in range(n)]
dp = [[0]*m for _ in range(n)]
visited = [[False] * m for _ in range(n)]

dfs(0, 0, 1)
print(maxcount)

👍 제출