LeetCode刷题：51. N 皇后（golang版）

• 题目描述

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

• 思路
  1、最重要的是斜线的判断条件，左下斜线为下标值和相等说明在一条斜线上，如（0，3）（1，2）（2，1）（3，0）这几个点在左下斜线。右下斜线为下标差值相等说明在一条斜线上，如（0，1）（1，2）（2，3）这几个点在右下斜线上
  2、使用深度优先递归，并根据条件剪枝处理
  3、used保存已被放置的点，true代表已被放置

• 代码

func solveNQueens(n int) [][]string {
	var (
		res = make([][]string, 0)
		used = make(map[int]map[int]bool)
	)
	//依次调用第0行的每一列
	for i:=0; i<n; i++ {
		used[0] = map[int]bool{i:true}
		dfs(n,0,i,used, &res)
		delete(used,i)
	}
	return res
}

//深度优先
func dfs(n int, row int, column int, used map[int]map[int]bool, res *[][]string){
	//递归终止条件，并满足题意，将结果转为题目要求的形式
	if row == n-1 {
		var s []string
		for i:=0; i<n; i++{
			var str string
			for j:=0; j<n; j++  {
				if used[i][j] == true {
					str += "Q"
				}else {
					str += "."
				}
			}
			s = append(s,str)
		}
		*res = append(*res,s)
		return
	}
	//判断新行的每一列
	for i:=0; i<n; i++ {
		if used[row+1][i] == false && judeg(used, row+1, i) {
			used[row+1] = map[int]bool{i : true}
			dfs(n,row+1,i,used,res)
			delete(used,row+1)
		}
	}

}
//判断该位置(willR,willC)能不能放，利用规则剪枝减少递归次数
func judeg(used map[int]map[int]bool, willR, willC int) bool {
	for row,v := range used {
		for column,_ := range v {
			//判断规则，左下线：下标和值相等，右下线：下标差值相等，列是否已使用过，满足任意一个即不合规，返回false
			if row+column == willR+willC || column-row == willC-willR || column == willC  {
				return false
			}
		}
	}
	//全部满足说明可以放置
	return true
}

• 测试代码

package main
import "fmt"

func main() {
	var k = 9
	res := solveNQueens(k)
	fmt.Println(res)
}