每日一练笔记|N皇后

题目：

N皇后问题 研究的是如何将 N个皇后放置在 N×N 的棋盘上，并且使皇后彼此之间不能相互攻击。
给你一个整数 N，返回所有不同的 N 皇后问题 的解决方案以及解决方案数。
每一种解法包含一个不同的 N 皇后问题 的棋子放置方案，该方案中 ‘Q’ 和 ‘#’ 分别代表了皇后和空位。
示例 1：

输入：N = 4
输出：[[“#Q##”,“###Q”,“Q###”,“##Q#”],[“##Q#”,“Q###”,“###Q”,“#Q##”]]
解释：如上图所示，4 皇后问题存在两个不同的解法。
示例 2：
输入：N = 1
输出：[[“Q”]]

提示：
1 <= N <= 9
皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

算法实现：

class Solution(object):
	def solveNQueens(self, n):
		if n == 0:
			return 0
		res = []
		board = [['#'] * n for t in range(n)]
		self.do_solveNQueens(res, board, n)
		return res
	def do_solveNQueens(self, res, board, num):
		if num == 0:
			res.append([''.join(t) for t in board])
			return
		ls = len(board)
		pos = ls - num
		check = [True] * ls
		for i in range(pos):
			for j in range(ls):
				if board[i][j] == 'Q':
					check[j] = False
					step = pos - i
					if j + step < ls:
						check[j + step] = False
					if j - step >= 0:
						check[j - step] = False
					break
		for j in range(ls):
			if check[j]:
				board[pos][j] = 'Q'
				self.do_solveNQueens(res, board, num - 1)
				board[pos][j] = '#'
if __name__ == '__main__':

	while(1):
		N = int(input('请输入N:'))
		if N>9 or N<=0:
			print('N的值输入错误，请重新输入：')
		else:
			break

	s = Solution()
	ans = s.solveNQueens(N)
	print ('解决方案有：', ans)
	print('解决方案数：', len(ans))