[BaekJoon] 1987 알파벳 (Gold 4) - Python

문제

 

세로 R칸, 가로 C칸으로 된 표 모양의 보드가 있다. 보드의 각 칸에는 대문자 알파벳이 하나씩 적혀 있고, 좌측 상단 칸 (1행 1열) 에는 말이 놓여 있다.

말은 상하좌우로 인접한 네 칸 중의 한 칸으로 이동할 수 있는데, 새로 이동한 칸에 적혀 있는 알파벳은 지금까지 지나온 모든 칸에 적혀 있는 알파벳과는 달라야 한다. 즉, 같은 알파벳이 적힌 칸을 두 번 지날 수 없다.

좌측 상단에서 시작해서, 말이 최대한 몇 칸을 지날 수 있는지를 구하는 프로그램을 작성하시오. 말이 지나는 칸은 좌측 상단의 칸도 포함된다.

---

입출력 예시

 

입력: 첫째 줄에 R과 C가 빈칸을 사이에 두고 주어진다. (1 ≤ R,C ≤ 20) 둘째 줄부터 R개의 줄에 걸쳐서 보드에 적혀 있는 C개의 대문자 알파벳들이 빈칸 없이 주어진다.

출력: 첫째 줄에 말이 지날 수 있는 최대의 칸 수를 출력한다.

 

입력예시1)

2 4
CAAB
ADCB

출력예시1)

3

 

입력예시2)

3 6
HFDFFB
AJHGDH
DGAGEH

출력예시2)

6

 

입력예시3)

5 5
IEFCJ
FHFKC
FFALF
HFGCF
HMCHH

출력예시3)

10

---

코드

 

import sys
input = sys.stdin.readline

R, C = map(int, input().rstrip().split())
grid = [list(input().rstrip()) for _ in range(R)]
d = [(0, 1), (0, -1), (1, 0), (-1, 0)]
visit = [False for _ in range(26)]
visit[ord(grid[0][0]) - 65] = True
ans = 0

def dfs(x, y, depth):
    
    global ans
    ans = max(ans, depth)

    for dx, dy in d:
        nx = x + dx
        ny = y + dy
        if 0 <= nx < R and 0 <= ny < C and not visit[ord(grid[nx][ny]) - 65]:
            visit[ord(grid[nx][ny]) - 65] = True
            dfs(nx, ny, depth + 1)
            visit[ord(grid[nx][ny]) - 65] = False

dfs(0, 0, 1)
print(ans)

---

실행 화면

 

---

채점 결과