[Python/BOJ] 1987. 알파벳

https://www.acmicpc.net/problem/1987

1987번: 알파벳

문제

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

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

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

입력

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

출력

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

풀이

Code

import sys
input = sys.stdin.readline

# 1. dfs 함수 정의
def dfs(x, y, string) :
    global max_len
    # 1-1.
    for dir_x, dir_y in dirs :
        nx, ny = x + dir_x, y + dir_y

        # 예외처리
        if nx < 0 or nx >= n or ny < 0 or ny >= m : continue
        # 방문한 적이 있는 경우 최댓값 업데이트
        if visited[nx][ny] : max_len = max(max_len, len(string))
        # 방문한 적이 없는 경우
        else :
            # 해당 문자가 문자열에 이미 있는 경우 최댓값 업데이트
            if board[nx][ny] in string : max_len = max(max_len, len(string))
            # 해당 문자가 문자열에 없는 경우
            else :
                # 방문 처리
                visited[nx][ny] = True
                # dfs 실행
                dfs(nx, ny, string + board[nx][ny])
                # 방문 취소
                visited[nx][ny] = False
                
n, m = map(int, input().split())
board = [list(map(str, input().rstrip())) for _ in range(n)]
dirs = [(-1, 0), (1, 0), (0, -1), (0, 1)]
string = board[0][0]
max_len = -int(1e9)
# 2. 방문 여부 리스트 생성
visited = [[False] * m for _ in range(n)]
visited[0][0] = True
# 3. dfs 실행
dfs(0, 0, string)
# 4. 결과 출력
print(max_len)