用JavaScript解决八皇后问题

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()

参考资料：

刘汝佳老师的算法竞赛入门经典（第二版）