《leetcode-go》n-皇后 ii

最新推荐文章于 2025-10-02 00:01:36 发布

原创最新推荐文章于 2025-10-02 00:01:36 发布 · 111 阅读

0 ·

CC 4.0 BY-SA版权

leetCode-go 专栏收录该内容

81 篇文章

订阅专栏

该博客探讨了如何计算n皇后问题的解决方案数量。通过递归地放置皇后并检查冲突，程序可以在不记录具体位置的情况下统计解的数量。核心函数`totalNQueens`和`getQueensNum`实现了这一逻辑，对于每一行，检查所有列是否可以放置皇后，并递归处理下一行。这种方法简化了问题，只关注解的数量而非具体的解。

继续思考“n-queens”问题

这次我们不是输出皇后的排列情况，而是输出n皇后问题一共有多少种解法

思路：这个更容易一点，不需要记录具体皇后的位置来返回，只需要记录临时的皇后位置，最后判断是否+1即可

package main

/**
  * 
  * @param n int整型 
  * @return int整型
*/
func totalNQueens(n int) int {
	// write code here
	num := 0
	getQueensNum(nil, 0, n, &num)
	return num
}

//找第i行皇后的位置，tmp临时的皇后位置，queens最终的皇后位置结果
func getQueensNum(tmp [][]int, i int, n int, num *int) {
	//如果已经找到n个就写入最终结果
	if i >= n {
		*num++
		return
	}
	for j := 0; j < n; j++ {
		//判断i,j是否可用
		flag := true
		if tmp != nil {
			for _, item := range tmp {
				//如果存在一行一列还有斜率为1，直接认为不可用
				if i == item[0] || j == item[1] || j-item[1] == i-item[0] || j-item[1] == -(i-item[0]) {
					flag = false
					break
				}
			}
		}
		//如果可用，找下一层
		if flag {
			newTmp := append([][]int{}, tmp...)
			getQueensNum(append(newTmp, []int{i, j}), i+1, n, num)
		}
	}
}