[Baekjoon] 1978번 알파벳

문제

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

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

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

입력

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

출력

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

예제

Example 1:

Input: 
2 4
CAAB
ADCB
Output: 
3

조건

시간 제한 : 2초

메모리 제한 : 256 MB

풀이과정

내 풀이

deque를 이용해 풀었는데 계속 시간 초과가 나왔다. 질문에 있는 글을 참고하여 set로 바꾸어 푸니 정답이 통과되었다.

import sys
input = sys.stdin.readline

# 세로 r칸, 가로 c칸
r, c = map(int, input().split())

# 그래프 입력받기
graph = []
for _ in range(r):
    graph.append(list(input()))

# 이동
dx = [-1, 0, 1, 0]
dy = [0, -1, 0, 1]

ans = 1
# BFS 진행
q = set([(0, 0, graph[0][0])])

while q:
    nx, ny, dist = q.pop()
    for i in range(4):
        nnx = nx + dx[i]
        nny = ny + dy[i]

        if nnx <= -1 or nny <= -1 or nnx >= r or nny >= c:
            continue

        if graph[nnx][nny] not in dist:
            q.add((nnx, nny, dist + graph[nnx][nny]))
            ans = max(ans, len(dist) + 1)

print(ans)