백준 알고리즘 2210번 숫자판 점프(python)

브루트포스 알고리즘 & DFS 문제이다. 문자열 s를 계속 업데이트하다가 길이가 6이면 set(res)에 추가해주고,

마지막에 set의 길이를 출력해주면 된다.

import sys
sys.setrecursionlimit(3000000)

def dfs(y, x, s):
    s += str(graph[y][x])
    if len(s) == 6:
        res.add(s)
        return ;
    for dy, dx in d:
        Y, X = y+dy, x+dx
        if (0 <= Y < 5) and (0 <= X < 5):
            dfs(Y, X, s)
    
graph = [list(map(int, input().split())) for _ in range(5)]
d = [(-1, 0), (1, 0), (0, -1), (0, 1)]
res = set()
for i in range(5):
    for j in range(5):
        s = ""
        dfs(i, j, s)
print(len(res))