1. import scala.collection.mutable.Map
  2. import scala.collection.mutable.Set
  4. case class Field(name:String, a: Int, b:Int, c:Int, d:Int){
  5. def good(x: Int) = (x >= a && x <= b) || (x >= c && x <= d)
  6. }
  8. object Field {
  9. def apply(line: String) : Field = {
  10. val Array(name, values) = line.split(": ")
  11. val Array(a,b,c,d) = values.split(" or ").map(_.split("-")) .flatten.map(_.toInt)
  12. Field(name, a, b, c, d)
  13. }
  14. }
  16. val C = io.Source.fromFile("input").getLines.toList
  18. val fields = C.takeWhile(_.size>0).map(Field.apply)
  19. val mine = C.dropWhile(_.size>0).drop(2).head.split(",").map(_.toInt).toArray
  20. val nearBy= C.dropWhile(_.size>0).drop(5).map(_.split(",").map(_.toInt)).map(_.toList)
  23. val goods = Set[Int]()
  25. for(x <- 1 to 1000) {
  26. if(fields.exists(f => f.good(x))) goods.add(x)
  27. }
  28. val validNear = nearBy.filter{ near =>
  29. near.filterNot(goods).isEmpty
  30. }
  32. var order : List[(Int, List[Field])] = Nil
  34. for(col <- 0 until nearBy.head.size){
  35. val vertValues = validNear.map(_.drop(col).head)
  36. val chosenFields = fields.filter(f => vertValues.forall(f.good))
  37. println(col+ " => " + chosenFields.map(_.name).size)
  38. order = (col, chosenFields) :: order
  39. }
  41. val check = order.sortBy(_._2.size)
  43. val woozy = check.map(o => (o._1, o._2.map(_.name).toSet))
  45. val result = Map[Int, String]()
  46. val found = Set[String]()
  48. woozy.foreach {
  49. case (idx, wuss) => {
  50. val curr = wuss.diff(found)
  51. result(idx) = curr.head
  52. found.add(curr.head)
  53. }
  54. }
  56. val recipe = result.filter {
  57. case (k, v) => v.contains("departure")
  58. }
  60. println("Recipe " + recipe)
  61. val woah = recipe.keys.map(k => mine(k)).map(_.toLong).product
  63. println(woah)