1. 背景
在8 x 8
的的国际棋盘中,放置八个皇后,要求任意皇后之间不能处于同一行,同一列,同对角线,有多少种放置方法
2. 思路
经过思考,不难发现,每行每列恰好各放置一个皇后
如果用map[row][col]
表示第row
行第col
列,则问题就变成了全排列的问
题
比较巧妙的地方是如何去判断当前皇后与之前的皇后是否冲突,
行不用判断,因为我们是逐行放置,列的判断比较容易
对角线的判断比较巧妙,
这里用的图是刘汝佳老师书上的图,图中可以看出,
主对角线上的行列相加的和都是相等的
副对角线上行列相减的插都是相等的
所以判断条件就是i + j == r + c || i - j == r - c
i和j指之前皇后的行与列,r和c指当前皇后的行与列
3. 实现代码
const N = 8
var map = []
var count = 0
// 初始化map数组
var init = function() {
for (let i = 0; i < N; i++) {
map[i] = []
for (let j = 0; j < N; j++) {
map[i][j] = 0
}
}
}
// 判断与前面的皇后是否冲突
var check = function(r, c) {
for (let i = 0; i < r; i++) {
for (let j = 0; j < N; j++) {
if (map[i][j] == 1) {
if (c == j || i + j == r + c || i - j == r - c) {
return false
}
}
}
}
return true
}
// 显示打印结果
var display = function() {
let str = ''
for (let i = 0; i < N; i++) {
for (let j = 0; j < N; j++) {
str = str + map[i][j] + ' '
}
str += '\n'
}
console.log(str)
}
// 按行搜索
var dfs = function(r) {
if (r == N) {
count++
display()
}
for (let c = 0; c < N; c++) {
if (check(r, c)) {
map[r][c] = 1
dfs(r + 1)
map[r][c] = 0
}
}
}
var __main = function() {
init()
dfs(0)
console.log('八皇后一共有:' + count + '种解法')
}
__main()
但其实我们不需要使用二维数组来标记行列,
更好的办法是用一维数组,数组的下标代表行,值代表列
一维数组实现代码:
const N = 8
var map = []
var count = 0
// 判断与前面的皇后是否冲突
var check = function(r) {
for (let i = 0; i < r; i++) {
if (
map[i] == map[r] ||
i + map[i] == r + map[r] ||
i - map[i] == r - map[r]
) {
return false
}
}
return true
}
// 显示打印结果
var display = function() {
let str = ''
for (let i = 0; i < N; i++) {
for (let j = 0; j < N; j++) {
if (map[i] == j) {
str += '1 '
} else {
str += '0 '
}
}
str += '\n'
}
console.log(str)
}
// 按行搜索
var dfs = function(r) {
if (r == N) {
count++
display()
}
for (let c = 0; c < N; c++) {
map[r] = c
if (check(r)) {
dfs(r + 1)
}
}
}
var __main = function() {
dfs(0)
console.log('八皇后一共有:' + count + '种解法')
}
__main()
参考资料:
刘汝佳老师的算法竞赛入门经典(第二版)