From 553969f0fdf6ed3ed9d017e5761b4cdee24506fb Mon Sep 17 00:00:00 2001
From: Evgeniy Khramtsov <xramtsov@gmail.com>
Date: Wed, 22 Jul 2009 06:52:47 +0000
Subject: [PATCH] delete/2 now does not crash when there is nothing to delete.
 fold/1 added

SVN Revision: 2385
---
 src/treap.erl | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/treap.erl b/src/treap.erl
index ab5c4d33f..360f543de 100644
--- a/src/treap.erl
+++ b/src/treap.erl
@@ -32,7 +32,8 @@
 	 delete_root/1,
 	 get_root/1,
 	 lookup/2,
-	 is_empty/1]).
+	 is_empty/1,
+	 fold/3]).
 
 empty() ->
     nil.
@@ -105,6 +106,8 @@ delete(Key, Tree) ->
     HashKey = {erlang:phash2(Key), Key},
     delete1(HashKey, Tree).
 
+delete1(_HashKey, nil) ->
+    nil;
 delete1(HashKey, {HashKey1, Priority1, Value1, Left, Right} = Tree) ->
     if
 	HashKey < HashKey1 ->
@@ -162,3 +165,9 @@ lookup1({HashKey1, Priority1, Value1, Left, Right}, HashKey) ->
 	    {ok, Priority1, Value1}
     end.
 
+fold(_F, Acc, nil) ->
+    Acc;
+fold(F, Acc, {{_Hash, Key}, Priority, Value, Left, Right}) ->
+    Acc1 = F({Key, Priority, Value}, Acc),
+    Acc2 = fold(F, Acc1, Left),
+    fold(F, Acc2, Right).