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