EightQueens

scala/src/main/scala/ch39_back_tracking/EightQueens.scala

@@ -0,0 +1,78 @@
+package ch39_back_tracking
+
+import scala.util.control.Breaks._
+
+class EightQueens {
+
+  //use array index to identify the row,the value of the row to identify the column
+  val result = new Array[Int](8)
+  var count = 0
+
+  def calc8Queues(row: Int): Unit = {
+    if (row == 8) {
+      //everything is done
+      print8Queens()
+      return
+    }
+
+    for (column <- Range(0, 8)) {
+      if (isOkOnColumn(row, column)) {
+        result(row) = column //place the column value into the array
+        calc8Queues(row + 1) //calculate next row
+      }
+    }
+
+  }
+
+  def isOkOnColumn(row: Int, column: Int): Boolean = {
+    var ok = true
+    var leftUp = column - 1
+    var rightUp = column + 1
+
+    breakable {
+      //will compare all the rows above current row
+      for (i <- row - 1 to 0 by -1) {
+        //check current column
+        if (result(i) == column) {
+          ok = false
+          break
+        }
+        //check left up
+        if (leftUp >= 0) {
+          if (result(i) == leftUp) {
+            ok = false
+            break
+          }
+        }
+        //check right up
+        if (rightUp < 8) {
+          if (result(i) == rightUp) {
+            ok = false
+            break
+          }
+        }
+        //move leftUp and rightUp
+        leftUp -= 1
+        rightUp += 1
+      }
+    }
+
+    ok
+  }
+
+  def print8Queens(): Unit = {
+    count +=1
+    for (row <- Range(0, 8)) {
+      for (column <- Range(0, 8)) {
+        if (result(row) == column) {
+          print("Q ")
+        } else {
+          print("* ")
+        }
+      }
+      //new line for next row
+      println("")
+    }
+    println(count+"==============")
+  }
+}

scala/src/test/scala/ch39_back_tracking/EightQueensTest.scala

@@ -0,0 +1,14 @@
+package ch39_back_tracking
+
+import org.scalatest.FlatSpec
+
+class EightQueensTest extends FlatSpec {
+
+  behavior of "EightQueensTest"
+
+  it should "calc8Queues" in {
+    val eightQueens = new EightQueens()
+    eightQueens.calc8Queues(0)
+  }
+
+}