{-# LANGUAGE CPP #-}
module BinomialQueue.Min (
MinQueue,
empty,
null,
size,
findMin,
getMin,
deleteMin,
deleteFindMin,
minView,
singleton,
insert,
union,
unions,
(!!),
take,
drop,
splitAt,
takeWhile,
dropWhile,
span,
break,
filter,
partition,
mapMaybe,
mapEither,
map,
foldrAsc,
foldlAsc,
foldrDesc,
foldlDesc,
toList,
toAscList,
toDescList,
fromList,
fromAscList,
fromDescList,
mapU,
foldrU,
foldlU,
foldlU',
foldMapU,
elemsU,
toListU,
seqSpine
) where
import Prelude hiding (null, take, drop, takeWhile, dropWhile, splitAt, span, break, (!!), filter, map)
import Data.Foldable (foldl')
import Data.Maybe (fromMaybe)
#if MIN_VERSION_base(4,9,0)
import Data.Semigroup (Semigroup((<>)))
#endif
import qualified Data.List as List
import BinomialQueue.Internals
#ifdef __GLASGOW_HASKELL__
import GHC.Exts (build)
#else
build :: ((a -> [a] -> [a]) -> [a] -> [a]) -> [a]
build f = f (:) []
#endif
findMin :: Ord a => MinQueue a -> a
findMin :: forall a. Ord a => MinQueue a -> a
findMin = a -> Maybe a -> a
forall a. a -> Maybe a -> a
fromMaybe ([Char] -> a
forall a. HasCallStack => [Char] -> a
error [Char]
"Error: findMin called on empty queue") (Maybe a -> a) -> (MinQueue a -> Maybe a) -> MinQueue a -> a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. MinQueue a -> Maybe a
forall a. Ord a => MinQueue a -> Maybe a
getMin
deleteMin :: Ord a => MinQueue a -> MinQueue a
deleteMin :: forall a. Ord a => MinQueue a -> MinQueue a
deleteMin MinQueue a
q = case MinQueue a -> Maybe (a, MinQueue a)
forall a. Ord a => MinQueue a -> Maybe (a, MinQueue a)
minView MinQueue a
q of
Maybe (a, MinQueue a)
Nothing -> MinQueue a
forall a. MinQueue a
empty
Just (a
_, MinQueue a
q') -> MinQueue a
q'
deleteFindMin :: Ord a => MinQueue a -> (a, MinQueue a)
deleteFindMin :: forall a. Ord a => MinQueue a -> (a, MinQueue a)
deleteFindMin = (a, MinQueue a) -> Maybe (a, MinQueue a) -> (a, MinQueue a)
forall a. a -> Maybe a -> a
fromMaybe ([Char] -> (a, MinQueue a)
forall a. HasCallStack => [Char] -> a
error [Char]
"Error: deleteFindMin called on empty queue") (Maybe (a, MinQueue a) -> (a, MinQueue a))
-> (MinQueue a -> Maybe (a, MinQueue a))
-> MinQueue a
-> (a, MinQueue a)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. MinQueue a -> Maybe (a, MinQueue a)
forall a. Ord a => MinQueue a -> Maybe (a, MinQueue a)
minView
(!!) :: Ord a => MinQueue a -> Int -> a
MinQueue a
q !! :: forall a. Ord a => MinQueue a -> Int -> a
!! Int
n | Int
n Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
>= MinQueue a -> Int
forall a. MinQueue a -> Int
size MinQueue a
q
= [Char] -> a
forall a. HasCallStack => [Char] -> a
error [Char]
"Data.PQueue.Min.!!: index too large"
MinQueue a
q !! Int
n = MinQueue a -> [a]
forall a. Ord a => MinQueue a -> [a]
toAscList MinQueue a
q [a] -> Int -> a
forall a. HasCallStack => [a] -> Int -> a
List.!! Int
n
{-# INLINE takeWhile #-}
takeWhile :: Ord a => (a -> Bool) -> MinQueue a -> [a]
takeWhile :: forall a. Ord a => (a -> Bool) -> MinQueue a -> [a]
takeWhile a -> Bool
p = (a -> Bool) -> [a] -> [a]
forall a. (a -> Bool) -> [a] -> [a]
List.takeWhile a -> Bool
p ([a] -> [a]) -> (MinQueue a -> [a]) -> MinQueue a -> [a]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. MinQueue a -> [a]
forall a. Ord a => MinQueue a -> [a]
toAscList
dropWhile :: Ord a => (a -> Bool) -> MinQueue a -> MinQueue a
dropWhile :: forall a. Ord a => (a -> Bool) -> MinQueue a -> MinQueue a
dropWhile a -> Bool
p = MinQueue a -> MinQueue a
drop' where
drop' :: MinQueue a -> MinQueue a
drop' MinQueue a
q = case MinQueue a -> Maybe (a, MinQueue a)
forall a. Ord a => MinQueue a -> Maybe (a, MinQueue a)
minView MinQueue a
q of
Just (a
x, MinQueue a
q') | a -> Bool
p a
x -> MinQueue a -> MinQueue a
drop' MinQueue a
q'
Maybe (a, MinQueue a)
_ -> MinQueue a
q
span :: Ord a => (a -> Bool) -> MinQueue a -> ([a], MinQueue a)
span :: forall a. Ord a => (a -> Bool) -> MinQueue a -> ([a], MinQueue a)
span a -> Bool
p MinQueue a
queue = case MinQueue a -> Maybe (a, MinQueue a)
forall a. Ord a => MinQueue a -> Maybe (a, MinQueue a)
minView MinQueue a
queue of
Just (a
x, MinQueue a
q')
| a -> Bool
p a
x -> let ([a]
ys, MinQueue a
q'') = (a -> Bool) -> MinQueue a -> ([a], MinQueue a)
forall a. Ord a => (a -> Bool) -> MinQueue a -> ([a], MinQueue a)
span a -> Bool
p MinQueue a
q' in (a
x a -> [a] -> [a]
forall a. a -> [a] -> [a]
: [a]
ys, MinQueue a
q'')
Maybe (a, MinQueue a)
_ -> ([], MinQueue a
queue)
break :: Ord a => (a -> Bool) -> MinQueue a -> ([a], MinQueue a)
break :: forall a. Ord a => (a -> Bool) -> MinQueue a -> ([a], MinQueue a)
break a -> Bool
p = (a -> Bool) -> MinQueue a -> ([a], MinQueue a)
forall a. Ord a => (a -> Bool) -> MinQueue a -> ([a], MinQueue a)
span (Bool -> Bool
not (Bool -> Bool) -> (a -> Bool) -> a -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> Bool
p)
{-# INLINE take #-}
take :: Ord a => Int -> MinQueue a -> [a]
take :: forall a. Ord a => Int -> MinQueue a -> [a]
take Int
n = Int -> [a] -> [a]
forall a. Int -> [a] -> [a]
List.take Int
n ([a] -> [a]) -> (MinQueue a -> [a]) -> MinQueue a -> [a]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. MinQueue a -> [a]
forall a. Ord a => MinQueue a -> [a]
toAscList
drop :: Ord a => Int -> MinQueue a -> MinQueue a
drop :: forall a. Ord a => Int -> MinQueue a -> MinQueue a
drop Int
n MinQueue a
queue = Int
n Int -> MinQueue a -> MinQueue a
forall a b. a -> b -> b
`seq` case MinQueue a -> Maybe (a, MinQueue a)
forall a. Ord a => MinQueue a -> Maybe (a, MinQueue a)
minView MinQueue a
queue of
Just (a
_, MinQueue a
queue')
| Int
n Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> Int
0 -> Int -> MinQueue a -> MinQueue a
forall a. Ord a => Int -> MinQueue a -> MinQueue a
drop (Int
n Int -> Int -> Int
forall a. Num a => a -> a -> a
- Int
1) MinQueue a
queue'
Maybe (a, MinQueue a)
_ -> MinQueue a
queue
splitAt :: Ord a => Int -> MinQueue a -> ([a], MinQueue a)
splitAt :: forall a. Ord a => Int -> MinQueue a -> ([a], MinQueue a)
splitAt Int
n MinQueue a
queue = Int
n Int -> ([a], MinQueue a) -> ([a], MinQueue a)
forall a b. a -> b -> b
`seq` case MinQueue a -> Maybe (a, MinQueue a)
forall a. Ord a => MinQueue a -> Maybe (a, MinQueue a)
minView MinQueue a
queue of
Just (a
x, MinQueue a
queue')
| Int
n Int -> Int -> Bool
forall a. Ord a => a -> a -> Bool
> Int
0 -> let ([a]
xs, MinQueue a
queue'') = Int -> MinQueue a -> ([a], MinQueue a)
forall a. Ord a => Int -> MinQueue a -> ([a], MinQueue a)
splitAt (Int
n Int -> Int -> Int
forall a. Num a => a -> a -> a
- Int
1) MinQueue a
queue' in (a
x a -> [a] -> [a]
forall a. a -> [a] -> [a]
: [a]
xs, MinQueue a
queue'')
Maybe (a, MinQueue a)
_ -> ([], MinQueue a
queue)
filter :: Ord a => (a -> Bool) -> MinQueue a -> MinQueue a
filter :: forall a. Ord a => (a -> Bool) -> MinQueue a -> MinQueue a
filter a -> Bool
p = (a -> Maybe a) -> MinQueue a -> MinQueue a
forall b a. Ord b => (a -> Maybe b) -> MinQueue a -> MinQueue b
mapMaybe (\a
x -> if a -> Bool
p a
x then a -> Maybe a
forall a. a -> Maybe a
Just a
x else Maybe a
forall a. Maybe a
Nothing)
partition :: Ord a => (a -> Bool) -> MinQueue a -> (MinQueue a, MinQueue a)
partition :: forall a.
Ord a =>
(a -> Bool) -> MinQueue a -> (MinQueue a, MinQueue a)
partition a -> Bool
p = (a -> Either a a) -> MinQueue a -> (MinQueue a, MinQueue a)
forall b c a.
(Ord b, Ord c) =>
(a -> Either b c) -> MinQueue a -> (MinQueue b, MinQueue c)
mapEither (\a
x -> if a -> Bool
p a
x then a -> Either a a
forall a b. a -> Either a b
Left a
x else a -> Either a a
forall a b. b -> Either a b
Right a
x)
map :: Ord b => (a -> b) -> MinQueue a -> MinQueue b
map :: forall b a. Ord b => (a -> b) -> MinQueue a -> MinQueue b
map a -> b
f = (a -> MinQueue b -> MinQueue b)
-> MinQueue b -> MinQueue a -> MinQueue b
forall a b. (a -> b -> b) -> b -> MinQueue a -> b
foldrU (b -> MinQueue b -> MinQueue b
forall a. Ord a => a -> MinQueue a -> MinQueue a
insert (b -> MinQueue b -> MinQueue b)
-> (a -> b) -> a -> MinQueue b -> MinQueue b
forall b c a. (b -> c) -> (a -> b) -> a -> c
. a -> b
f) MinQueue b
forall a. MinQueue a
empty
{-# INLINE toList #-}
toList :: Ord a => MinQueue a -> [a]
toList :: forall a. Ord a => MinQueue a -> [a]
toList = MinQueue a -> [a]
forall a. Ord a => MinQueue a -> [a]
toAscList
foldlDesc :: Ord a => (b -> a -> b) -> b -> MinQueue a -> b
foldlDesc :: forall a b. Ord a => (b -> a -> b) -> b -> MinQueue a -> b
foldlDesc = (a -> b -> b) -> b -> MinQueue a -> b
forall a b. Ord a => (a -> b -> b) -> b -> MinQueue a -> b
foldrAsc ((a -> b -> b) -> b -> MinQueue a -> b)
-> ((b -> a -> b) -> a -> b -> b)
-> (b -> a -> b)
-> b
-> MinQueue a
-> b
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (b -> a -> b) -> a -> b -> b
forall a b c. (a -> b -> c) -> b -> a -> c
flip
{-# INLINE fromDescList #-}
fromDescList :: [a] -> MinQueue a
fromDescList :: forall a. [a] -> MinQueue a
fromDescList [a]
xs = (MinQueue a -> a -> MinQueue a) -> MinQueue a -> [a] -> MinQueue a
forall b a. (b -> a -> b) -> b -> [a] -> b
forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl' ((a -> MinQueue a -> MinQueue a) -> MinQueue a -> a -> MinQueue a
forall a b c. (a -> b -> c) -> b -> a -> c
flip a -> MinQueue a -> MinQueue a
forall a. a -> MinQueue a -> MinQueue a
insertMinQ') MinQueue a
forall a. MinQueue a
empty [a]
xs
elemsU :: MinQueue a -> [a]
elemsU :: forall a. MinQueue a -> [a]
elemsU = MinQueue a -> [a]
forall a. MinQueue a -> [a]
toListU