[leetcode-in-go] 0052-N-Queens II

shida_csdn

于 2019-07-17 21:57:48 发布

阅读量184

点赞数

分类专栏： Golang 经典题目文章标签： leetcode N-Queues II

本文链接：https://blog.youkuaiyun.com/shida_csdn/article/details/96370331

版权

Golang 同时被 2 个专栏收录

51 篇文章

订阅专栏

经典题目

47 篇文章

订阅专栏

本文探讨了N皇后问题，这是一个经典的计算机科学问题，目标是在N×N的棋盘上放置N个皇后，使得没有任何两个皇后在同一行、同一列或同一对角线上。文章详细介绍了使用回溯法解决此问题的方法，并提供了具体的Go语言实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.

Given an integer n, return the number of distinct solutions to the n-queens puzzle.

Example:

Input: 4
Output: 2
Explanation: There are two distinct solutions to the 4-queens puzzle as shown below.
[
[".Q…", // Solution 1
“…Q”,
“Q…”,
“…Q.”],

["…Q.", // Solution 2
“Q…”,
“…Q”,
“.Q…”]
]

解题思路

回溯

func available(current []int, index int) bool {
	n := len(current)
	for i, v := range current {
		// 同一列
		if index == v {
			return false
		}

		// 对角线
		if (n-i) == index-v || (n-i) == v-index {
			return false
		}
	}

	return true
}

func core(current []int, n int, cnt *int) {
	if len(current) == n {
        *cnt += 1
	}

	for i := 0; i < n; i++ {
		if available(current, i) {
			core(append(current, i), n, cnt)
		}
	}
}

func totalNQueens(n int) int {
    cnt := 0
	if n == 0 {
		return cnt
	}
	core([]int{}, n, &cnt)
	return cnt
}