progs/knight3_sol.scala
changeset 45 8399976b77fe
child 50 9891c9fac37e
equal deleted inserted replaced
44:9cfa37c91444 45:8399976b77fe
       
     1 // Part 2 about finding a single tour for a board
       
     2 //================================================
       
     3 
       
     4 
       
     5 type Pos = (Int, Int)
       
     6 type Path = List[Pos]
       
     7 
       
     8 def print_board(dim: Int, path: Path): Unit = {
       
     9   println
       
    10   for (i <- 0 until dim) {
       
    11     for (j <- 0 until dim) {
       
    12       print(f"${path.reverse.indexOf((i, j))}%3.0f ")
       
    13     }
       
    14     println
       
    15   } 
       
    16 }
       
    17 
       
    18 def add_pair(x: Pos)(y: Pos): Pos = 
       
    19   (x._1 + y._1, x._2 + y._2)
       
    20 
       
    21 def is_legal(dim: Int, path: Path)(x: Pos): Boolean = 
       
    22   0 <= x._1 && 0 <= x._2 && x._1 < dim && x._2 < dim && !path.contains(x)
       
    23 
       
    24 def moves(x: Pos): List[Pos] = 
       
    25   List(( 1,  2),( 2,  1),( 2, -1),( 1, -2),
       
    26        (-1, -2),(-2, -1),(-2,  1),(-1,  2)).map(add_pair(x))
       
    27 
       
    28 def legal_moves(dim: Int, path: Path, x: Pos): List[Pos] = 
       
    29   moves(x).filter(is_legal(dim, path))
       
    30 
       
    31 def ordered_moves(dim: Int, path: Path, x: Pos): List[Pos] = 
       
    32   legal_moves(dim, path, x).sortBy((x) => legal_moves(dim, path, x).length)
       
    33 
       
    34 
       
    35 def first(xs: List[Pos], f: Pos => Option[Path]): Option[Path] = xs match {
       
    36   case Nil => None
       
    37   case x::xs => {
       
    38     val result = f(x)
       
    39     if (result.isDefined) result else first(xs, f)
       
    40   }
       
    41 }
       
    42 
       
    43 
       
    44 def first_closed_tour_heuristics(dim: Int, path: Path): Option[Path] = {
       
    45   if (path.length == dim * dim && moves(path.head).contains(path.last)) Some(path)
       
    46   else
       
    47     first(ordered_moves(dim, path, path.head), (x: Pos) => first_closed_tour_heuristics(dim, x::path))
       
    48 }
       
    49 
       
    50 for (dim <- 1 to 7) {
       
    51   val t = first_closed_tour_heuristics(dim, List((dim / 2, dim / 2)))
       
    52   println(s"${dim} x ${dim} closed: " + (if (t == None) "" else { print_board(dim, t.get) ; "" }))
       
    53 }
       
    54 
       
    55 
       
    56 def first_tour_heuristics(dim: Int, path: Path): Option[Path] = {
       
    57   if (path.length == dim * dim) Some(path)
       
    58   else
       
    59     first(ordered_moves(dim, path, path.head), (x: Pos) => first_tour_heuristics(dim, x::path))
       
    60 }
       
    61 
       
    62 for (dim <- 1 to 50) {
       
    63   val t = first_tour_heuristics(dim, List((dim / 2, dim / 2)))
       
    64   println(s"${dim} x ${dim}: " + (if (t == None) "" else { print_board(dim, t.get) ; "" }))
       
    65 }