scala 解决8皇后问题

来源:互联网 发布:软件系统性能指标 编辑:程序博客网 时间:2024/06/03 09:00
val n = 14val 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分钟

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

0 0
原创粉丝点击