Problem:
４つの整数の和 II

4,000 以下の正の整数 n を入力し、0 〜 1000 の範囲の整数 a, b, c, d の組で

a + b + c + d = n

を満たすものの組み合わせ数を出力するプログラムを作成して下さい。

Input

複数のデータセットが与えられます。各データセットに n が１行に与えられます。入力の最後まで処理して下さい。

データセットの数は 50 を超えません。

Output

各データセットごとに、a, b, c, d の組み合わせの個数を１行に出力して下さい。

Sample Input

2
3
35

Output for the Sample Input

10
20
8436

Solution in Scala (judge verdict: Wrong Answer)

import scala.io.Source.stdin
import scala.math.{max,min}

object Main extends App {

  val p = (
      for (i <- (0 to 2000) ++ (1999 to 0 by -1)) yield (i+3)*(i+2)*(i+1)/6 - max(2*(i-1000)*(i-999)*(i-998)/3,0)
  )

  stdin.getLines.foreach { x => println(p(x.toInt)) }
}