go语言刷题：52. N皇后 II

申明：本文只用做自己的学习记录

题目

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给一个整数 n ，返回 n 皇后问题不同的解决方案的数量。
示例 1：
输入：n = 4
输出：2
解释：如上图所示，4 皇后问题存在两个不同的解法。
示例 2：

输入：n = 1
输出：1

代码

func totalNQueens(n int) (ans int) {
	columns := make([]bool, n)        	// 列上是否有皇后
	diagonals1 := make([]bool, 2*n-1) 	// 左上到右下是否有皇后
	diagonals2 := make([]bool, 2*n-1) 	// 右上到左下是否有皇后
	row := 0							//初始化行的值
	backtrack(n, &ans, row, &columns, &diagonals1, &diagonals2)
	return ans
}

func backtrack(n int, ans *int, row int, columns *[]bool, diagonals1 *[]bool, diagonals2 *[]bool) {
	//1.判断成立条件，如果这个树状搜索能够一直搜索下去
	//当搜索到第n-1时，条件都成立，则该方法可用，所以当返回n时就不需要再计算了，已经没有值了
	//故当row==n时,可用方法+1
	if row == n {
		*ans++
		return
	}
	//2.在row行的基础上，进行列遍历
	for colIndex, isHasQueen := range *columns {
		//d1为左对角线索引；d2为右对角线索引，都是基于行、列索引的基础上定位该点的左、右对角线，并向后遍历
		d1, d2 := row+(n-1)-colIndex, row+colIndex
		//3.减枝
		//如果该列存在皇后||左对角线存在皇后||右对角线存在皇后，则跳过本条支线的搜索，不再继续,
		if isHasQueen || (*diagonals1)[d1] || (*diagonals2)[d2] {
			continue
		}
		//如果不存在上述情况，则可以进行下一步遍历
		//将该点的列、左对角线、右对角线状态设置为存在皇后的状态
		(*columns)[colIndex] = true
		(*diagonals1)[d1] = true
		(*diagonals2)[d2] = true
		//进行下一步搜索，行+1
		backtrack(n, ans, row+1, columns, diagonals1, diagonals2)
		//4.回溯，恢复到之前的状态进行下一次，基于row的列搜索
		(*columns)[colIndex] = false
		(*diagonals1)[d1] = false
		(*diagonals2)[d2] = false
	}
	return
}