From c89eda2baf2b0ea37fff3fc86a5f49ddd25b7c3c Mon Sep 17 00:00:00 2001
From: Philippe Suter <philippe.suter@gmail.com>
Date: Wed, 7 Jul 2010 17:47:52 +0000
Subject: [PATCH]

---
 testcases/BinarySearchTree.scala | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/testcases/BinarySearchTree.scala b/testcases/BinarySearchTree.scala
index 0e7f56f48..a01ccc3f6 100644
--- a/testcases/BinarySearchTree.scala
+++ b/testcases/BinarySearchTree.scala
@@ -20,9 +20,13 @@ object BinarySearchTree {
 
     def contains(tree: Tree, value: Int) : Boolean = tree match {
         case Leaf() => false
-        case Node(_, v, _) if v == value => true 
-        case Node(l, v, r) if v < value => contains(r, value)
-        case Node(l, v, r) if v > value => contains(l, value)
+        case n @ Node(l, v, r) => if(v < value) {
+          contains(r, value)
+        } else if(v > value) {
+          contains(l, value)
+        } else {
+          true
+        }
     }
 
     def contents(tree: Tree) : Set[Int] = tree match {
-- 
GitLab