Euler scala Ex12

Ex12.scala

/*
  1부터 n까지의 자연수를 차례로 더하여 구해진 값을 삼각수라고 합니다.
  예를 들어 7번째 삼각수는 1 + 2 + 3 + 4 + 5 + 6 + 7 = 28이 됩니다.
  이런 식으로 삼각수를 구해 나가면 다음과 같습니다.

  1, 3, 6, 10, 15, 21, 28, 36, 45, 55, ...
  이 삼각수들의 약수를 구해봅시다.

   1: 1
   3: 1, 3
   6: 1, 2, 3, 6
   10: 1, 2, 5, 10
   15: 1, 3, 5, 15
   21: 1, 3, 7, 21
   28: 1, 2, 4, 7, 14, 28
   위에서 보듯이, 5개 이상의 약수를 갖는 첫번째 삼각수는 28입니다.

   그러면 500개 이상의 약수를 갖는 가장 작은 삼각수는 얼마입니까?
*/
object Ex12 extends App {
  val result = Stream.from(1).map(n => (1 to n).sum).find(n => countDivisors(n) > 500).get

  println(result)

  def countDivisors(n: Int): Int = {
    var count = 0
    var end = Math.sqrt(n).toInt
    for (val i <- 1 to end) {
      if (n % i == 0) count += 2
    }
    if (end * end == n) count = count + 1

    return count
  }
}