marking3/knight1_test3b.scala
author Christian Urban <urbanc@in.tum.de>
Sun, 09 Dec 2018 01:36:49 +0000
changeset 243 9bb36426c781
child 244 a359976a6f3e
permissions -rw-r--r--
updated
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
243
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     1
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     2
type Pos = (Int, Int)    // a position on a chessboard 
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     3
type Path = List[Pos]    // a path...a list of positions
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     4
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     5
def count_all_tours_urban(dim: Int) = {
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     6
  for (i <- (0 until dim).toList; 
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     7
       j <- (0 until dim).toList) yield count_tours(dim, List((i, j)))
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     8
}
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     9
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    10
def add_pair_urban(x: Pos)(y: Pos): Pos = 
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    11
  (x._1 + y._1, x._2 + y._2)
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    12
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    13
def is_legal_urban(dim: Int, path: Path)(x: Pos): Boolean = 
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    14
  0 <= x._1 && 0 <= x._2 && x._1 < dim && x._2 < dim && !path.contains(x)
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    15
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    16
def moves_urban(x: Pos): List[Pos] = 
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    17
  List(( 1,  2),( 2,  1),( 2, -1),( 1, -2),
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    18
       (-1, -2),(-2, -1),(-2,  1),(-1,  2)).map(add_pair_urban(x))
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    19
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    20
def legal_moves_urban(dim: Int, path: Path, x: Pos): List[Pos] = 
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    21
  moves_urban(x).filter(is_legal_urban(dim, path))
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    22
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    23
def correct_urban(dim: Int)(p: Path): Boolean = p match {
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    24
  case Nil => true
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    25
  case x::Nil => true
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    26
  case x::y::p => if (legal_moves_urban(dim, p, y).contains(x)) correct_urban(dim)(y::p) else false
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    27
}
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    28
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    29
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    30
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    31
val ts_urban = enum_tours(5, List((0, 2)))
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    32
assert(ts_urban.map(correct_urban(5)).forall(_ == true) == true)
9bb36426c781 updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    33
assert(ts_urban.length == 56)