import java.text.SimpleDateFormat
import java.util.Calendar
class CMagicNumber(n:Int) {
val m:Int = n * n
val a:Int = n * (n * n + 1) / 2
def calc:Int = {
def _calc(p:Int,c:Int,s:Int,r:List[Int],g:Int):Int = r match {
case List() => g + 1
case _ => c match {
case 0 => _calc(r.head,1,r.head,r.tail, g)
case _ => r.filter { d => d > p }.foldLeft(g) { (g,d) =>
((c == n - 1) && (s + d == a)) match {
case true => _calc(0,0,0,r.filter { e => e != d },g)
case _ => (d + s >= a) match {
case true => g
case _ => _calc(d,c + 1,s + d,r.filter { e => e != d },g)
}
}
}
}
}
_calc(0,0,0,(1 to m).toList,0)
}
}
object MagicNumber {
def printTime:Unit = println((new SimpleDateFormat("yyyy/MM/dd HH:mm:ss")).format(Calendar.getInstance.getTime))
def main(args:Array[String]):Unit = {
try {
printTime
printf("result:%d\n",(new CMagicNumber(args.length match { case 1 => args(0).toInt; case _ => 5 })).calc)
printTime
} catch {
case e => e.printStackTrace
}
}
}
lunlumo #7555() [ Scala ] Rating0/0=0.00
Pentium1.6GHzの端末で17分少し掛かりました。
import java.text.SimpleDateFormat import java.util.Calendar class CMagicNumber(n:Int) { val m:Int = n * n val a:Int = n * (n * n + 1) / 2 def calc:Int = { def _calc(p:Int,c:Int,s:Int,r:List[Int],g:Int):Int = r match { case List() => g + 1 case _ => c match { case 0 => _calc(r.head,1,r.head,r.tail, g) case _ => r.filter { d => d > p }.foldLeft(g) { (g,d) => ((c == n - 1) && (s + d == a)) match { case true => _calc(0,0,0,r.filter { e => e != d },g) case _ => (d + s >= a) match { case true => g case _ => _calc(d,c + 1,s + d,r.filter { e => e != d },g) } } } } } _calc(0,0,0,(1 to m).toList,0) } } object MagicNumber { def printTime:Unit = println((new SimpleDateFormat("yyyy/MM/dd HH:mm:ss")).format(Calendar.getInstance.getTime)) def main(args:Array[String]):Unit = { try { printTime printf("result:%d\n",(new CMagicNumber(args.length match { case 1 => args(0).toInt; case _ => 5 })).calc) printTime } catch { case e => e.printStackTrace } } }Rating0/0=0.00-0+
[ reply ]