----------------------------------------------------------------------------- -- | -- Module : Data.LCA.Online.Monoidal -- Copyright : (C) 2012 Edward Kmett -- License : BSD-style (see the file LICENSE) -- -- Maintainer : Edward Kmett <ekmett@gmail.com> -- Stability : experimental -- Portability : portable -- -- Provides online calculation of the the lowest common ancestor in /O(log h)/ -- by compressing the spine of the paths using a skew-binary random access -- list. -- -- Algorithms used here assume that the key values chosen for @k@ are -- globally unique. -- ---------------------------------------------------------------------------- module Data.LCA.Online.Monoidal ( Path , toList, fromList , map, mapHom, mapWithKey , traverse, traverseWithKey , empty , cons , uncons, view , null , length , measure , isAncestorOf , keep, mkeep , drop, mdrop , (~=) , lca, mlca ) where import Control.Applicative hiding (empty) import Data.Foldable hiding (toList) import Data.Monoid (Monoid(..)) import Prelude hiding (length, null, drop, map) import Data.LCA.View infixl 6 <> (<>) :: Monoid a => a -> a -> a (<>) = mappend {-# INLINE (<>) #-} -- | Complete binary trees -- -- NB: we could ensure the complete tree invariant data Tree a = Bin a {-# UNPACK #-} !Int a (Tree a) (Tree a) | Tip {-# UNPACK #-} !Int a deriving (Show, Read) instance Foldable Tree where foldMap f (Tip _ a) = f a foldMap f (Bin _ _ a l r) = f a <> foldMap f l <> foldMap f r measureT :: Tree a -> a measureT (Tip _ a) = a measureT (Bin a _ _ _ _) = a bin :: Monoid a => Int -> a -> Tree a -> Tree a -> Tree a bin n a l r = Bin (a <> measureT l <> measureT r) n a l r sameT :: Tree a -> Tree b -> Bool sameT xs ys = root xs == root ys where root (Tip k _) = k root (Bin _ k _ _ _) = k -- | A compressed 'Path' as a skew binary random access list data Path a = Nil | Cons a {-# UNPACK #-} !Int -- the number of elements @n@ in this entire skew list {-# UNPACK #-} !Int -- the number of elements @w@ in this binary tree node (Tree a) -- a complete binary tree @t@ of with @w@ elements (Path a) -- @n - w@ elements in a linked list @ts@, of complete trees in ascending order by size deriving (Show, Read) instance Foldable Path where foldMap _ Nil = mempty foldMap f (Cons _ _ _ t ts) = foldMap f t <> foldMap f ts -- | Extract a monoidal summary of a 'Path'. measure :: Monoid a => Path a -> a measure Nil = mempty measure (Cons a _ _ _ _) = a consT :: Monoid a => Int -> Tree a -> Path a -> Path a consT w t ts = Cons (measureT t <> measure ts) (w + length ts) w t ts consN :: Monoid a => Int -> Int -> Tree a -> Path a -> Path a consN n w t ts = Cons (measureT t <> measure ts) n w t ts -- | /O(n)/ Re-annotate a 'Path' full of monoidal values using a different 'Monoid'. map :: Monoid b => (a -> b) -> Path a -> Path b map f = go where go Nil = Nil go (Cons _ n k t ts) = consN n k (goT t) (go ts) goT (Tip k a) = Tip k (f a) goT (Bin _ k a l r) = bin k (f a) (goT l) (goT r) {-# INLINE map #-} -- | /O(n)/ Re-annotate a 'Path' full of monoidal values with access to the key. mapWithKey :: Monoid b => (Int -> a -> b) -> Path a -> Path b mapWithKey f = go where go Nil = Nil go (Cons _ n k t ts) = consN n k (goT t) (go ts) goT (Tip k a) = Tip k (f k a) goT (Bin _ k a l r) = bin k (f k a) (goT l) (goT r) {-# INLINE mapWithKey #-} -- | /O(n)/ Re-annotate a 'Path' full of monoidal values/ -- -- Unlike 'map', @'mapHom' f@ assumes that f is a 'Monoid' homomorphism, that is to say you must ensure -- -- @ -- f a `'mappend'` f b = f (a `'mappend'` b) -- f 'mempty' = 'mempty' -- @ mapHom :: (a -> b) -> Path a -> Path b mapHom f = go where go Nil = Nil go (Cons a n k t ts) = Cons (f a) n k (goT t) (go ts) goT (Tip k a) = Tip k (f a) goT (Bin m k a l r) = Bin (f m) k (f a) (goT l) (goT r) {-# INLINE mapHom #-} -- | Convert a 'Path' to a list of @(ID, value)@ pairs. toList :: Path a -> [(Int,a)] toList Nil = [] toList (Cons _ _ _ t ts) = go t (toList ts) where go (Tip k a) xs = (k,a) : xs go (Bin _ k a l r) xs = (k,a) : go l (go r xs) -- | Build a 'Path' from a list of @(ID, value)@ pairs. fromList :: Monoid a => [(Int,a)] -> Path a fromList [] = Nil fromList ((k,a):xs) = cons k a (fromList xs) -- | Traverse a 'Path' with access to the node IDs. traverseWithKey :: (Applicative f, Monoid b) => (Int -> a -> f b) -> Path a -> f (Path b) traverseWithKey f = go where go Nil = pure Nil go (Cons _ n k t ts) = consN n k <$> goT t <*> go ts goT (Tip k a) = Tip k <$> f k a goT (Bin _ k a l r) = bin k <$> f k a <*> goT l <*> goT r {-# INLINE traverseWithKey #-} -- | Traverse a 'Path' yielding a new monoidal annotation. traverse :: (Applicative f, Monoid b) => (a -> f b) -> Path a -> f (Path b) traverse f = go where go Nil = pure Nil go (Cons _ n k t ts) = consN n k <$> goT t <*> go ts goT (Tip k a) = Tip k <$> f a goT (Bin _ k a l r) = bin k <$> f a <*> goT l <*> goT r {-# INLINE traverse #-} -- | The empty 'Path' empty :: Path a empty = Nil {-# INLINE empty #-} -- | /O(1)/ Determine the 'length' of a 'Path'. length :: Path a -> Int length Nil = 0 length (Cons _ n _ _ _) = n {-# INLINE length #-} -- | /O(1)/ Returns 'True' iff the path is 'empty'. null :: Path a -> Bool null Nil = True null _ = False {-# INLINE null #-} -- | /O(1)/ Invariant: most operations assume that the keys @k@ are globally unique -- -- Extend the 'Path' with a new node ID and value. cons :: Monoid a => Int -> a -> Path a -> Path a cons k a (Cons m n w t (Cons _ _ w' t2 ts)) | w == w' = Cons (a <> m) (n + 1) (2 * w + 1) (bin k a t t2) ts cons k a ts = Cons (a <> measure ts) (length ts + 1) 1 (Tip k a) ts {-# INLINE cons #-} -- | /O(1)/ Extract the node ID and value from the newest node on the 'Path'. uncons :: Monoid a => Path a -> Maybe (Int, a, Path a) uncons Nil = Nothing uncons (Cons _ _ _ (Tip k a) ts) = Just (k, a, ts) uncons (Cons _ _ w (Bin _ k a l r) ts) = Just (k, a, consT w2 l (consT w2 r ts)) where w2 = div w 2 {-# INLINE uncons #-} -- | /O(1)/ Extract the node ID and value from the newest node on the 'Path', slightly faster than 'uncons'. view :: Monoid a => Path a -> View Path a view Nil = Root view (Cons _ _ _ (Tip k a) ts) = Node k a ts view (Cons _ _ w (Bin _ k a l r) ts) = Node k a (consT w2 l (consT w2 r ts)) where w2 = div w 2 {-# INLINE view #-} -- | /O(log (h - k))/ to keep @k@ elements of 'Path' of 'length' @h@, and provide a monoidal summary of the dropped elements mkeep :: Monoid a => Int -> Path a -> (a, Path a) mkeep = go mempty where go as _ Nil = (as, Nil) go as k xs@(Cons _ n w t ts) | k >= n = (as, xs) | otherwise = case compare k (n - w) of GT -> goT as (k - n + w) w t ts EQ -> (as <> measureT t, ts) LT -> go (as <> measureT t) k ts -- goT :: Monoid a => Int -> Int -> Tree a -> Path a -> Path a goT as n w (Bin _ _ a l r) ts = case compare n w2 of LT -> goT (as <> a <> measureT l) n w2 r ts EQ -> (as <> a <> measureT l, consT w2 r ts) GT | n == w - 1 -> (as <> a, consT w2 l (consT w2 r ts)) | otherwise -> goT (as <> a) (n - w2) w2 l (consT w2 r ts) where w2 = div w 2 goT as _ _ _ ts = (as, ts) {-# INLINE mkeep #-} -- | /O(log (h - k))/ to @'keep' k@ elements of 'Path' of 'length' @h@ keep :: Monoid a => Int -> Path a -> Path a keep k xs = snd (mkeep k xs) {-# INLINE keep #-} -- | /O(log k)/ to @'drop' k@ elements from a 'Path' drop :: Monoid a => Int -> Path a -> Path a drop k xs = snd (mdrop k xs) {-# INLINE drop #-} -- | /O(log k)/ to drop @k@ elements from a 'Path' and provide a monoidal summary of the dropped elements mdrop :: Monoid a => Int -> Path a -> (a, Path a) mdrop k xs = mkeep (length xs - k) xs {-# INLINE mdrop #-} -- | /O(log h)/ @xs `'isAncestorOf'` ys@ holds when @xs@ is a prefix starting at the root of path @ys@. isAncestorOf :: Monoid b => Path a -> Path b -> Bool isAncestorOf xs ys = xs ~= keep (length xs) ys infix 4 ~= -- | /O(1)/ Compare to see if two trees have the same leaf key (~=) :: Path a -> Path b -> Bool Nil ~= Nil = True Cons _ _ _ s _ ~= Cons _ _ _ t _ = sameT s t _ ~= _ = False -- | /O(log h)/ Compute the lowest common ancestor of two paths lca :: (Monoid a, Monoid b) => Path a -> Path b -> Path a lca xs ys = zs where (_, zs, _, _) = mlca xs ys -- | /O(log h)/ Compute the lowest common ancestor of two paths along with a monoidal summary of their respective tails. mlca :: (Monoid a, Monoid b) => Path a -> Path b -> (a, Path a, b, Path b) mlca xs0 ys0 = case compare nxs nys of LT -> let (bs, ys) = mkeep nxs ys0 in go mempty bs xs0 ys EQ -> go mempty mempty xs0 ys0 GT -> let (as, xs) = mkeep nys xs0 in go as mempty xs ys0 where nxs = length xs0 nys = length ys0 go as bs pa@(Cons _ _ w x xs) pb@(Cons _ _ _ y ys) | sameT x y = (as, pa, bs, pb) | xs ~= ys = goT as bs w x y xs ys | otherwise = go (as <> measureT x) (bs <> measureT y) xs ys go as bs _ _ = (as, Nil, bs, Nil) goT as bs w (Bin _ _ a la ra) (Bin _ _ b lb rb) pa pb | sameT la lb = (as <> a, consT w2 la (consT w2 ra pa), bs <> b, consT w2 lb (consT w2 rb pb)) | sameT ra rb = goT (as <> a) (bs <> b) w2 la lb (consT w ra pa) (consT w rb pb) | otherwise = goT (as <> a <> measureT la) (bs <> b <> measureT lb) w2 ra rb pa pb where w2 = div w 2 goT as bs _ ta tb pa pb = (as <> measureT ta, pa, bs <> measureT tb, pb) {-# INLINE mlca #-}