Main.scala

import scala.io.Source;

object Main {

  def main(args: Array[String]): Unit = {
    val lines = groupBlanksIterator(Source.fromFile("day6in.txt").getLines).toList

    val partone = lines.map(countGroup (_)).sum

    println(partone)

    val parttwo = lines.map(countGroup2(_)).sum

    println(parttwo)
  }

  def countGroup(group: List[String]) : Int = {
    group.flatten.toSet.size
  }

  def countGroup2(group: List[String])  = {
    group.flatten.groupBy(a => a).values.filter(b => b.size == group.size).size
  }

  def groupBlanksIterator(xs:Iterator[String]) =
    new Iterator[List[String]]
    { def hasNext = xs.hasNext; def next = xs.takeWhile(_.nonEmpty).toList}
}