main_templates4/knight4.scala
author Christian Urban <christian.urban@kcl.ac.uk>
Wed, 02 Nov 2022 21:49:42 +0000
changeset 430 274c865b3878
child 432 de701b64a4e0
permissions -rw-r--r--
updated
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
430
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     1
// Part 4 about finding a single tour on "mutilated" chessboards
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     2
//==============================================================
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     3
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     4
object M4d { // for preparing the jar
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     5
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     6
type Pos = (Int, Int)
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     7
type Path = List[Pos]
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     8
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     9
def print_board(dim: Int, path: Path): Unit = {
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    10
  println()
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    11
  for (i <- 0 until dim) {
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    12
    for (j <- 0 until dim) {
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    13
      print(f"${path.reverse.indexOf((i, j))}%4.0f ")
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    14
    }
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    15
    println()
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    16
  } 
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    17
}
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    18
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    19
def add_pair(x: Pos, y: Pos): Pos = 
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    20
  (x._1 + y._1, x._2 + y._2)
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    21
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    22
def is_legal(dim: Int, path: Path, x: Pos): Boolean = 
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    23
  0 <= x._1 && 0 <= x._2 && x._1 < dim && x._2 < dim && !path.contains(x)
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    24
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    25
def moves(x: Pos): List[Pos] = 
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    26
  List(( 1,  2),( 2,  1),( 2, -1),( 1, -2),
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    27
       (-1, -2),(-2, -1),(-2,  1),(-1,  2)).map(add_pair(x, _))
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    28
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    29
def legal_moves(dim: Int, path: Path, x: Pos): List[Pos] = 
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    30
  moves(x).filter(is_legal(dim, path, _))
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    31
 
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    32
import scala.annotation.tailrec
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    33
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    34
@tailrec
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    35
def first(xs: List[Pos], f: Pos => Option[Path]): Option[Path] = xs match {
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    36
  case Nil => None
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    37
  case x::xs => {
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    38
    val result = f(x)
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    39
    if (result.isDefined) result else first(xs, f)
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    40
  }
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    41
}
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    42
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    43
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    44
def one_tour_pred(dim: Int, path: Path, n: Int, pred: Pos => Boolean): Option[Path] = {
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    45
  if (path.length == n) Some(path)
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    46
  else
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    47
    first(legal_moves(dim, path, path.head).filter(pred), (x: Pos) => one_tour_pred(dim, x::path, n, pred))
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    48
}
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    49
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    50
//print_board(8, one_tour_pred(8, List((0, 0)), 40, x => x._1 < 5).get)
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    51
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    52
274c865b3878 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    53
}