Advent of Code 04 part 1.scala

import java.io.FileInputStream
import scala.io.StdIn

@main def solve4() =
  val validKeys: Set[String] = Set("byr", "iyr", "eyr", "hgt", "hcl", "ecl", "pid") 
  def isValid(passport: Vector[(String, String)]): Boolean = {
    val keys = passport.map { _._1 }.toSet
    (validKeys diff keys).isEmpty
  }

  val in = new FileInputStream("example4-1.in")
  System.setIn(in)

  val inputs = Iterator.continually(StdIn.readLine()).takeWhile(_ != null).toVector
  val passportElements = inputs.foldLeft(Vector[Vector[(String, String)]](Vector())) { (acc, line) =>
    if line.trim.isEmpty then
      acc :+ Vector()
    else
      acc.init :+ (acc.last ++ line.split(" ").map { x => 
        val Array(key, value) = x.split(":")
        (key, value)
      }.toVector) 
  }
  println(passportElements.count(isValid))