其他分享
首页 > 其他分享> > Leetcode 52: N皇后II

Leetcode 52: N皇后II

作者:互联网

题目描述:
n 皇后问题 研究的是如何将 n 个皇后放置在 n × n 的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n ,返回 n 皇后问题 不同的解决方案的数量。
示例:

  1. 输入:n = 4
    输出:2
  2. 输入:n = 1
    输出:1
    思路:
    Leetcode51一样的思路,只是不需要转换形式把结果输出来,只需统计结果的个数。
class Solution(object):
    def totalNQueens(self, n):
        """
        :type n: int
        :rtype: int
        """
        self.result = []
        self.put_queens(0, [], [], [], n, [])
        return len(self.result)

    def put_queens(self, level, col, pie, na, n, cur_state):
        if level == n:
            self.result.append(cur_state)
            return
        for i in range(n):
            if i not in col and i+level not in pie and i-level not in na:
                self.put_queens(level+1, col+[i], pie+[i+level], na+[i-level], n, cur_state+[i])

复杂度分析:
时间复杂度:O(N!),其中 N是皇后数量。
空间复杂度:O(N)

标签:level,self,pie,52,II,result,复杂度,皇后,Leetcode
来源: https://blog.csdn.net/coding_diamond/article/details/122048824