📩 출처
문제
세로 R칸, 가로 C칸으로 된 표 모양의 보드가 있다. 보드의 각 칸에는 대문자 알파벳이 하나씩 적혀 있고, 좌측 상단 칸 (1행 1열) 에는 말이 놓여 있다.
말은 상하좌우로 인접한 네 칸 중의 한 칸으로 이동할 수 있는데, 새로 이동한 칸에 적혀 있는 알파벳은 지금까지 지나온 모든 칸에 적혀 있는 알파벳과는 달라야 한다. 즉, 같은 알파벳이 적힌 칸을 두 번 지날 수 없다.
좌측 상단에서 시작해서, 말이 최대한 몇 칸을 지날 수 있는지를 구하는 프로그램을 작성하시오. 말이 지나는 칸은 좌측 상단의 칸도 포함된다.
입력
첫째 줄에 R과 C가 빈칸을 사이에 두고 주어진다. (1 ≤ R,C ≤ 20) 둘째 줄부터 R개의 줄에 걸쳐서 보드에 적혀 있는 C개의 대문자 알파벳들이 빈칸 없이 주어진다.
출력
첫째 줄에 말이 지날 수 있는 최대의 칸 수를 출력한다.
👉 생각
- BFS를 돌릴 때 cnt와 이때까지 알파벳이 있는 문자열을 q에 넣어준다.
- 이제까지와 다른건 deque를 사용한 것이 아니라 set 자료구조를 사용했다. deque의 시간복잡도는 O(n)인 반면 set의 시간복잡도는 o(1)이기 때문에 deque로 돌리면 시간초과가 발생했다!
import sys
def bfs(x, y):
q = set([])
q.add((x, y, arr[x][y], 1))
answer = 0
while q:
x, y, word, cnt = q.pop()
if cnt > answer:
answer = cnt
for dx, dy in [(-1,0), (1,0), (0,-1), (0,1)]:
nx, ny = x + dx, y + dy
if 0 <= nx < r and 0 <= ny < c and arr[nx][ny] not in word:
q.add((nx, ny, word+arr[nx][ny], cnt+1))
return answer
r, c = map(int, input().split())
arr = [list(sys.stdin.readline().rstrip()) for _ in range(r)]
print(bfs(0,0))
