Leetcode 52: N皇后II
作者:互联网
题目描述:
n 皇后问题 研究的是如何将 n 个皇后放置在 n × n 的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n ,返回 n 皇后问题 不同的解决方案的数量。
示例:
- 输入:n = 4
输出: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