分析

定義一個函數生成當前queens的結果
然后queens用來記錄每行的queen分別在什么位置
backtrack用來記錄當前第i行的queen的位置在哪
如果當前i = n就可以加入最終的結果了
否則開始遍歷每個j，如果當前的ij滿足cols diag1 和 diag2就可以取這個j
然后設定queen 并加入三個set
然后dfs下一層 再回溯為之前的set 再遍歷下一個j

ac code

class Solution:def solveNQueens(self, n: int) -> List[List[str]]:# backtrack + setrow = ['.'] * nqueens = [-1] * ncols, diag1, diag2 = set(), set(), set()ans = []def generateBoard():board = []for i in range(n):row[queens[i]] = 'Q'board.append(''.join(row))row[queens[i]] = '.'return boarddef backtrack(i):if i == n:ans.append(generateBoard())else:# next row which colfor j in range(n):if j in cols or i + j in diag1 or i - j in diag2:continue# dfs + backtrackqueens[i] = jcols.add(j)diag1.add(i + j)diag2.add(i - j)backtrack(i + 1)cols.remove(j)diag1.remove(i + j)diag2.remove(i - j)backtrack(0)return ans

總結

dfs + backtrack的復雜度并不是n的n次方 中間剪枝了不少的
復雜度是o(N!)

總結

以上是生活随笔為你收集整理的leetcode：面试题 08.12. 八皇后【dfs + backtrack】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。