0051-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
皇后彼此不能相互攻击,也就是说:任何两个皇后都不能处于同一条横行、纵行或斜线上。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-queens

参考:

  • https://leetcode-cn.com/problems/n-queens/solution/dai-ma-sui-xiang-lu-51-n-queenshui-su-fa-2k32/

python

# 0051.N皇后

class Solution:
    def solveNQueens(self, n: int) -> [[str]]:
        if not n:
            return
        board = [['.']*n for _ in range(n)]
        res = []

        def isVaild(board, row, col):
            # 判断同列是否冲突
            for i in range(len(board)):
                if board[i][col] == 'Q':
                    return False
            # 判断左上角是否冲突
            i = row - 1
            j = col - 1
            while i >= 0 and j >= 0:
                if board[i][j] == 'Q':
                    return False
                i -= 1
                j -= 1
            # 判断右上角是否冲突
            i = row - 1
            j = col + 1
            while i >= 0 and j < len(board):
                if board[i][j] == 'Q':
                    return False
                i -= 1
                j += 1
            return True

        def backTrack(board, row, n):
            # 如果走到最后一步,说明已经找到一个解
            if row == n:
                tmp_res = []
                for tmp in board:
                    tmp_str = ''.join(tmp)
                    tmp_res.append(tmp_str)
                res.append(tmp_res)
            for col in range(n):
                if not isVaild(board, row, col):
                    continue
                board[row][col] = 'Q'
                backTrack(board, row+1, n) # 同行不重复
                board[row][col] = '.'

        backTrack(board, 0, n)
        return res

golang

package backTrack

import "strings"

var res [][]string

func isVaild(board [][]string, row, col int) (res bool) {
	n := len(board)
	// 检查行
	for i:=0;i=0 && j >=0; i,j = i-1, j-1 {
		if board[i][j] == "Q" {
			return false
		}
	}
	// 检查右对角线
	for i,j:=row,col;i>=0 && j