leetcode:面试题 08.12. 八皇后【dfs + backtrack】
生活随笔
收集整理的這篇文章主要介紹了
leetcode:面试题 08.12. 八皇后【dfs + backtrack】
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
分析
定義一個函數生成當前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】的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 算法-回溯backtrack
- 下一篇: backtrack5 初步配置