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

l_mofi

于 2021-06-06 23:55:55 发布

阅读量178

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode刷题文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/admin_o1/article/details/117638953

LeetCode刷题专栏收录该内容

19 篇文章

订阅专栏

该博客探讨了如何使用深度优先搜索(DFS)解决经典的N皇后问题，计算放置N个皇后在N×N棋盘上而不相互攻击的所有不同解法数量。通过避免行、列和对角线上的冲突，递归地尝试每个位置并更新解法计数。代码中定义了用于记录已放置皇后的辅助数据结构，并展示了如何在搜索过程中检查冲突和回溯。

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

思路
1、和前一题一样，只是结果换成了有多少种解法，小改动下即可
代码

package main

import "fmt"

func main() {
	var k = 4
	res := totalNQueens(k)
	fmt.Println(res)
}


func totalNQueens(n int) int {
	var (
		sum int = 0
		used = make(map[int]map[int]bool)
	)
	for i:=0; i<n; i++ {
		used[0] = map[int]bool{i:true}
		dfs(n,0,i,used,&sum)
		delete(used,i)
	}
	return sum
}


func dfs(n int, row int, column int, used map[int]map[int]bool,sum *int){
	if row == n-1 {
		*sum ++
		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,sum)
			delete(used,row+1)
		}
	}

}

func judeg(used map[int]map[int]bool, willR, willC int) bool {
	for row,v := range used {
		for column,_ := range v {
			if row+column == willR+willC || column-row == willC-willR || column == willC  {
				return false
			}
		}
	}
	return true
}