알고리즘/DFS, BFS, 시뮬, 백트래킹

[프로그래머스] N-Queen 문제

VIPeveloper 2021. 11. 9. 23:44
728x90
반응형

문제 설명

가로, 세로 길이가 n인 정사각형으로된 체스판이 있습니다. 체스판 위의 n개의 퀸이 서로를 공격할 수 없도록 배치하고 싶습니다.

예를 들어서 n이 4인경우 다음과 같이 퀸을 배치하면 n개의 퀸은 서로를 한번에 공격 할 수 없습니다.

체스판의 가로 세로의 세로의 길이 n이 매개변수로 주어질 때, n개의 퀸이 조건에 만족 하도록 배치할 수 있는 방법의 수를 return하는 solution함수를 완성해주세요.

제한사항

  • 퀸(Queen)은 가로, 세로, 대각선으로 이동할 수 있습니다.
  • n은 12이하의 자연수 입니다.

입출력 예

nresult

4 2

입출력 예 설명

입출력 예 #1
문제의 예시와 같습니다.

 

오답노트

안되는 문제는 그냥 외워라! 그러면 이해하게 될 것이다! 는 말은 진리인 것 같다. 처음에 정말 모르겠어서 그냥 외웠다.

몇번 따라 쓰다보니까 아 이래서 이렇게 코딩했구나! 알게 되었던 문제였다.

참 어려운 하루였다~

def dfs(queen,row,n):
    count = 0
    if row == n:
        return 1
    for col in range(n):
        queen[row] = col
        for i in range(row):
            if queen[row] == queen[i]:
                break
            if abs(queen[row] - queen[i]) == row - i:
                break
        else:
            count += dfs(queen,row+1,n)
    return count


print(dfs([0]*8, 0, 8))
print(dfs([0]*4,0,4))

 

728x90
반응형