首页 > itarticle > leetcode_n

leetcode_n

admin 11月 13, 2020 0

N-Queens

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.
(N皇后的问题，求所有解)

Given an integer n, return all distinct solutions to the n-queens puzzle. Each solution contains a distinct board configuration of the n-queens’ placement, where ‘Q’ and ‘.’ both indicate a queen and an empty space respectively.

Example:

1. 回溯法

类似于这种解空间可以用树表示，且存在某种约束条件的问题可以用回溯法求解。具体实现过程如下：

import copy

class :
	def check(self, matrix, _i, _j):
		for i in range(self.n):
			for j in range(self.n):
				if matrix[i][j] == 'Q':
					if j == _j or abs(_i - i) == abs(_j - j):
						return False
		return True

	def dfs(self, i, matrix):
		if (i == self.n):
			result = []
			for row in matrix:
				result.append(''.join(row))
			self.result.append(result)
		else:
			for j in range(self.n):
				if self.check(matrix, i, j):
					matrix[i][j] = 'Q'
					self.dfs(i+1, matrix)
					matrix[i][j] = '.'
	
	def solveNQueens(self, n):
		"""
		:type n: int
		:rtype: List[List[str]]
		"""
		self.n = n
		self.result = []

		matrix = [['.' for _ in range(self.n)] for _ in range(self.n)]

		self.dfs(0, matrix)
		return self.result