Scala example source code file (binarytrees.scala-3.scala)

This example Scala source code file (binarytrees.scala-3.scala) is included in the DevDaily.com "Java Source Code Warehouse" project. The intent of this project is to help you "Learn Java by Example" ^TM.

Java - Scala tags/keywords

array, array, int, int, string, string, tree, tree

The Scala binarytrees.scala-3.scala source code

/* The Computer Language Shootout
   http://shootout.alioth.debian.org/

   contributed by Kannan Goundan
   modified by Isaac Gouy
*/

object binarytrees {
   def main(args: Array[String]) = {
      val n = try { Integer.parseInt(args(0)) } catch { case _ => 1 }
      val minDepth = 4
      val maxDepth = Math.max(minDepth+2,n)

      print("stretch tree", maxDepth+1, new Tree(0,maxDepth+1).isum)

      val longLivedTree = new Tree(0,maxDepth)

      var depth = minDepth
      while (depth <= maxDepth) {
         val iterations = 1 << (maxDepth - depth + minDepth)

         var sum = 0
         var i = 1
         while (i <= iterations) {
            sum = sum + new Tree(i,depth).isum + new Tree(-i,depth).isum
            i = i + 1
         }
         print(iterations*2 + "\t trees", depth, sum)

         depth = depth + 2
      }

      print("long lived tree", maxDepth, longLivedTree.isum)
   }

   def print(name: String, depth: Int, check: Int) =
      Console.println(name + " of depth " + depth + "\t check: " + check)
}


final class Tree(_i: Int, _left: Tree, _right: Tree) {
    private val i = _i
    private var left = _left
    private var right = _right

    def this(i: Int, depth: Int) = {
       this(i, null, null)
       if (depth > 0){ 
          left = new Tree((2*i)-1, depth-1)
          right = new Tree(2*i, depth-1)
       }
    }

    def isum(): Int = i + (if (left == null) 0 else left.isum - right.isum)
}