progs/sudoku_test.scala
author Christian Urban <christian.urban@kcl.ac.uk>
Tue, 07 Dec 2021 01:35:00 +0000
changeset 414 78cb8e9a9d17
parent 383 72d6a4af4b4a
permissions -rw-r--r--
updated
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
383
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     1
// Sudoku
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     2
//========
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     3
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     4
// call parallel version with
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     5
//
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     6
// scala -cp scala-parallel-collections_2.13-0.2.0.jar sudoku_test.scala 
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     7
//
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     8
// or
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     9
//
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    10
// scalac -cp scala-parallel-collections_2.13-0.2.0.jar sudoku_test.scala
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    11
// java -cp .:scala-library-2.13.0.jar:scala-parallel-collections_2.13-0.2.0.jar Sudoku
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    12
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    13
object Sudoku extends App { 
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    14
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    15
import scala.collection.parallel.CollectionConverters._
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    16
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    17
type Pos = (Int, Int)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    18
val emptyValue = '.'
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    19
val maxValue = 9
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    20
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    21
val allValues = "123456789".toList
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    22
val indexes = (0 to 8).toList
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    23
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    24
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    25
def empty(game: String) = game.indexOf(emptyValue)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    26
def isDone(game: String) = empty(game) == -1 
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    27
def emptyPosition(game: String) = {
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    28
  val e = empty(game)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    29
  (e % maxValue, e / maxValue)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    30
}
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    31
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    32
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    33
def get_row(game: String, y: Int) = indexes.map(col => game(y * maxValue + col))
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    34
def get_col(game: String, x: Int) = indexes.map(row => game(x + row * maxValue))
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    35
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    36
def get_box(game: String, pos: Pos): List[Char] = {
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    37
    def base(p: Int): Int = (p / 3) * 3
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    38
    val x0 = base(pos._1)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    39
    val y0 = base(pos._2)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    40
    for (x <- (x0 until x0 + 3).toList;
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    41
         y <- (y0 until y0 + 3).toList) yield game(x + y * maxValue)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    42
}         
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    43
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    44
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    45
def update(game: String, pos: Int, value: Char): String = 
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    46
  game.updated(pos, value)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    47
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    48
def toAvoid(game: String, pos: Pos): List[Char] = 
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    49
  (get_col(game, pos._1) ++ get_row(game, pos._2) ++ get_box(game, pos))
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    50
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    51
def candidates(game: String, pos: Pos): List[Char] = 
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    52
  allValues.diff(toAvoid(game, pos))
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    53
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    54
def search(game: String): List[String] = {
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    55
  if (isDone(game)) List(game)
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    56
  else {
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    57
    val cs = candidates(game, emptyPosition(game))
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    58
    cs.par.map(c => search(update(game, empty(game), c))).toList.flatten
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    59
  }
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    60
}
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    61
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    62
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    63
def pretty(game: String): String = 
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    64
  "\n" + (game.grouped(maxValue).mkString(",\n"))
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    65
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    66
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    67
val game2 = """8........
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    68
              |..36.....
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    69
              |.7..9.2..
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    70
              |.5...7...
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    71
              |....457..
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    72
              |...1...3.
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    73
              |..1....68
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    74
              |..85...1.
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    75
              |.9....4..""".stripMargin.replaceAll("\\n", "")
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    76
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    77
// for measuring time
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    78
def time_needed[T](i: Int, code: => T) = {
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    79
  val start = System.nanoTime()
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    80
  for (j <- 1 to i) code
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    81
  val end = System.nanoTime()
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    82
  s"${(end - start) / 1.0e9} secs"
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    83
}
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    84
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    85
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    86
println(time_needed(10, search(game2)))
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    87
72d6a4af4b4a updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    88
}