scala 解决8皇后问题

val n = 14
val Col = new Array[Boolean](n)
val Bevel1 = new Array[Boolean](n*2)
val Bevel2 = new Array[Boolean](n*2)
def col(y: Int) = !Col(y)
def bevel1(x: Int, y: Int) = !Bevel1(x-y+n-1)
def bevel2(x: Int, y: Int) = !Bevel2(x+y)
def dfs(x: Int): Int = if(x == n) 1 else Range(0, n).foldLeft(0)((a, b) => a + solve(x, b))
def solve(x: Int, i: Int): Int = {
    if(col(i) && bevel2(x, i) && bevel1(x, i)){
        Col(i) = true; Bevel1(x-i+n-1) = true; Bevel2(x+i) = true
        val ans = dfs(x+1)
        Col(i) = false; Bevel1(x-i+n-1) = false; Bevel2(x+i) = false
        ans
    }else 0
}
println("The number of queue is: " + n + " " + "result: " + dfs(0))

这个代码解决15后大约需要半分钟,16后3分钟

思路: 回溯。当走到某个地方后,先判断这个地方能不能走,就是判断该位置所在的行,列,斜是否有皇后。以为是一行一行往下走,该行上肯定没有皇后

你可能感兴趣的:(scala 解决8皇后问题)