Counting Sheep.scala

import java.io.FileOutputStream
import java.io.FileInputStream

import scala.io.StdIn

object Main extends App {

    Console.setIn(new FileInputStream("example.in"))
    
    Console.setIn(new FileInputStream("A-small-attempt0.in"))
    Console.setOut(new FileOutputStream("A-small-attempt0.out"))

    Console.setIn(new FileInputStream("A-large-practice.in"))
    Console.setOut(new FileOutputStream("A-large-practice.out"))


    def solve(n:Int): Option[Long] = n match {
        case 0 => None
        case _ => Stream.iterate((0, Set[Char]())){
                case (v, chars) => (v + n, chars ++ (v + n).toString)
            }.find(_._2.size == 10).map(_._1)
    }

    val cases = StdIn.readLine().toInt
    (1 to cases) foreach { i => {
        val input = StdIn.readLine().toInt
        println(s"Case #$i: ${solve(input).map(_.toString).getOrElse("INSOMNIA")}")
    }}
}