[코테] 백준 브루트 포스 1987번 문제

1987번 : 알파벳

문제 링크

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

문제 설명

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

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

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

입력

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

출력

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

문제풀이

dfs를 진행하면 되는데 시간초과에서 많이 애먹었다. pypy3로 제출하면 해결되며 처음 문자열을 입력받을 때부터 정수형으로 변경하는 등 재귀 구문 안에서 최대한 함수 호출을 적게하도록 변경하면 시간 내에 제출 가능했다.

import sys
input = sys.stdin.readline

R, C = map(int, input().split())
board = [list(map(lambda x:ord(x)-65, input())) for _ in range(R)]    # 입력받은 문자를 정수형으로 변경
visited = [False] * 26    # 방문 정보 저장하는 배열
dy = [1, -1, 0, 0]
dx = [0, 0, 1, -1]
result = 1

def back(y, x, cnt) :
    global result
    result = max(result, cnt)
    
    for i in range(4) : 
        ny, nx = y+dy[i], x+dx[i]
        
        # 만약 방문하지 않은 알파벳이면 재귀 진행
        if 0 <= ny < R and 0 <= nx < C and not visited[board[ny][nx]] : 
            visited[board[ny][nx]] = True
            back(ny, nx, cnt+1)
            visited[board[ny][nx]] = False

visited[board[0][0]] = True    # 좌측 상단칸은 무조건 방문하므로 방문 표시 
back(0,0,1)
print(result)