ALGO: add goLeft/goRight in KDTree

2014-11-29 23:49:39 +01:00
parent 543b08df2c
commit 26221f3046
1 changed files with 9 additions and 0 deletions
--- a/Algorithms/KDTree/KDTree.hs
+++ b/Algorithms/KDTree/KDTree.hs
@@ -168,3 +168,12 @@ kdTreeToRoseTree (KTNode ln val _ rn) =
  Node (show . unp2 $ val) [kdTreeToRoseTree ln, kdTreeToRoseTree rn]


+goLeft :: KDTree a -> Maybe (KDTree a)
+goLeft (KTNode ln _ _ _) = Just ln
+goLeft _                 = Nothing
+
+
+goRight :: KDTree a -> Maybe (KDTree a)
+goRight (KTNode _ _ _ rn) = Just rn
+goRight _                 = Nothing
+