diff options
| author | aarne <aarne@cs.chalmers.se> | 2008-05-20 11:47:44 +0000 |
|---|---|---|
| committer | aarne <aarne@cs.chalmers.se> | 2008-05-20 11:47:44 +0000 |
| commit | 31bf84122b21efb444aa8d055472e166ffb90783 (patch) | |
| tree | 1f051909336f1534346bcccde8dda59beab02f64 /src-2.9/GF/Data/RedBlack.hs | |
| parent | 74f048dcf41de3540778de54dfa7541fa5b39c46 (diff) | |
moved all old source code to src-2.9 ; src will be for GF 3 development
Diffstat (limited to 'src-2.9/GF/Data/RedBlack.hs')
| -rw-r--r-- | src-2.9/GF/Data/RedBlack.hs | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/src-2.9/GF/Data/RedBlack.hs b/src-2.9/GF/Data/RedBlack.hs new file mode 100644 index 000000000..fd70dba63 --- /dev/null +++ b/src-2.9/GF/Data/RedBlack.hs @@ -0,0 +1,64 @@ +---------------------------------------------------------------------- +-- | +-- Module : RedBlack +-- Maintainer : Markus Forsberg +-- Stability : Stable +-- Portability : Haskell 98 +-- +-- > CVS $Date: 2005/04/21 16:22:07 $ +-- > CVS $Author: bringert $ +-- > CVS $Revision: 1.6 $ +-- +-- Modified version of Osanaki's implementation. +----------------------------------------------------------------------------- + +module GF.Data.RedBlack ( + emptyTree, + isEmpty, + Tree, + lookupTree, + insertTree, + flatten + ) where + +data Color = R | B + deriving (Show,Read) + +data Tree key el = E | T Color (Tree key el) (key,el) (Tree key el) + deriving (Show,Read) + +balance :: Color -> Tree a b -> (a,b) -> Tree a b -> Tree a b +balance B (T R (T R a x b) y c) z d = T R (T B a x b) y (T B c z d) +balance B (T R a x (T R b y c)) z d = T R (T B a x b) y (T B c z d) +balance B a x (T R (T R b y c) z d) = T R (T B a x b) y (T B c z d) +balance B a x (T R b y (T R c z d)) = T R (T B a x b) y (T B c z d) +balance color a x b = T color a x b + +emptyTree :: Tree key el +emptyTree = E + +isEmpty :: Tree key el -> Bool +isEmpty (E) = True +isEmpty _ = False + +lookupTree :: Ord a => a -> Tree a b -> Maybe b +lookupTree _ E = Nothing +lookupTree x (T _ a (y,z) b) + | x < y = lookupTree x a + | x > y = lookupTree x b + | otherwise = return z + +insertTree :: Ord a => (a,b) -> Tree a b -> Tree a b +insertTree (key,el) tree = T B a y b + where + T _ a y b = ins tree + ins E = T R E (key,el) E + ins (T color a y@(key',el') b) + | key < key' = balance color (ins a) y b + | key > key' = balance color a y (ins b) + | otherwise = T color a (key',el) b + +flatten :: Tree a b -> [(a,b)] +flatten E = [] +flatten (T _ left (key,e) right) + = (flatten left) ++ ((key,e):(flatten right)) |
