path: root/sources/examples/patterns.scala

package examples;

object patterns {

  trait Tree;
  case class Branch(left: Tree, right: Tree) extends Tree;
  case class Leaf(x: Int) extends Tree;

  val tree1 = Branch(Branch(Leaf(1), Leaf(2)), Branch(Leaf(3), Leaf(4)));

  def sumLeaves(t: Tree): Int = t match {
    case Branch(l, r) => sumLeaves(l) + sumLeaves(r)
    case Leaf(x) => x
  }

  def find[a,b](it: Iterator[Pair[a, b]], x: a): Option[b] = {
    var result: Option[b] = _;
    while (it.hasNext && result == null) {
      val Pair(x1, y) = it.next;
      if (x == x1) result = Some(y)
    }
    if (result == null) None else result
  }

  def printFinds[a](xs: List[Pair[a, String]], x: a) =
    find(xs.elements, x) match {
      case Some(y) => System.out.println(y)
      case None => System.out.println("no match")
    }

  def main(args: Array[String]): Unit = {
    Console.println("sum of leafs=" + sumLeaves(tree1));
    printFinds(List(Pair(3, "three"), Pair(4, "four")), 4)
  }
}