[프로그래머스] N-Queen (Python)

https://programmers.co.kr/learn/courses/30/lessons/12952

코딩테스트 연습 - N-Queen

다른 풀이를 참고했다. 아직 dfs를 짜는데 좀 익숙하지 않은 것 같다.

 

나중에 다시 살펴봐야겠다.

import sys
sys.setrecursionlimit(10**6)
def dfs(queen,n,row):
    answer = 0
    if n == row:
        return 1
    
    for i in range(n):
        queen[row] = i
        
        for j in range(row):
            if queen[j] == queen[row]:
                # 같은 열에 있을 경우
                break
            if abs(queen[j] - queen[row]) == row-j:
                # 대각선인 경우
                break
        else:
            answer += dfs(queen,n,row+1)
    return answer

def solution(n):
    queen = [0]*n
    answer = dfs(queen,n,0)
    return answer