分析

定義一個函數(shù)生成當(dāng)前queens的結(jié)果
然后queens用來記錄每行的queen分別在什么位置
backtrack用來記錄當(dāng)前第i行的queen的位置在哪
如果當(dāng)前i = n就可以加入最終的結(jié)果了
否則開始遍歷每個j，如果當(dāng)前的ij滿足cols diag1 和 diag2就可以取這個j
然后設(shè)定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

總結(jié)

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。