Modulenonempty-containers-0.4.0.0Haskell2010
Data.IntMap.NonEmpty.Strict
Non-Empty Finite Integer-Indexed Maps (strict interface)
The NEIntMap v type represents a non-empty finite map (sometimes
called a dictionary) from integer keys to values of type v.
An NEIntMap is strict in its keys and values.
See documentation for NEIntMap for information on how to convert and manipulate such non-empty maps.
This module essentially re-imports the API of Data.IntMap.Strict and its IntMap type, along with semantics and asymptotics. In most situations, asymptotics are different only by a constant factor. In some situations, asmyptotics are even better (constant-time instead of log-time).
Because NEIntMap is implemented using IntMap, all of the caveats of using IntMap apply (such as the limitation of the maximum size of maps).
All functions take non-empty maps as inputs. In situations where their results can be guarunteed to also be non-empty, they also return non-empty maps. In situations where their results could potentially be empty, IntMap is returned instead.
Some variants of functions (like alter', alterF', adjustMin, adjustMax, adjustMinWithKey, adjustMaxWithKey) are provided in a way restructured to preserve guaruntees of non-empty maps being returned.
Some functions (like mapEither, partition, split) have modified return types to account for possible configurations of non-emptiness.
This module is intended to be imported qualified, to avoid name clashes with Prelude and Data.IntMap functions:
import qualified Data.IntMap.NonEmpty.Strict as NEIMNote that all asmyptotics O(f(n)) in this module are actually
O(min(W, f(n))), where W is the number of bits in an Int (32 or
64). That is, if f(n) is greater than W, all operations are
constant-time.
Import Data.IntMap.NonEmpty.Lazy for a variant lazy in values.
- 2 types
- 126 values
- Packagenonempty-containers-0.4.0.0
- Exports130
- LanguageHaskell2010
- LicenceBSD-3-Clause
- SourceStrict.hs
Non-Empty IntMap Type
2 declarationsConversions between empty and non-empty maps
O(1) match, O(log n) usage of contents. The IsNonEmpty and
IsEmpty patterns allow you to treat a IntMap as if it were either
a IsNonEmpty n (where n is a NEIntMap) or an IsEmpty.
For example, you can pattern match on a IntMap:
myFunc :: IntMap K X -> Y
myFunc (IsNonEmpty n) = -- here, the user provided a non-empty map, and n is the NEIntMap
myFunc IsEmpty = -- here, the user provided an empty map.
Matching on IsNonEmpty n means that the original IntMap was not
empty, and you have a verified-non-empty NEIntMap n to use.
Note that patching on this pattern is O(1). However, using the contents requires a O(log n) cost that is deferred until after the pattern is matched on (and is not incurred at all if the contents are never used).
A case statement handling both IsNonEmpty and IsEmpty provides complete coverage.
This is a bidirectional pattern, so you can use IsNonEmpty to convert a NEIntMap back into a IntMap, obscuring its non-emptiness (see toMap).
O(1). The IsNonEmpty and IsEmpty patterns allow you to treat
a IntMap as if it were either a IsNonEmpty n (where n is
a NEIntMap) or an IsEmpty.
Matching on IsEmpty means that the original IntMap was empty.
A case statement handling both IsNonEmpty and IsEmpty provides complete coverage.
This is a bidirectional pattern, so you can use IsEmpty as an
expression, and it will be interpreted as Data.IntMap.empty.
See IsNonEmpty for more information.
O(log n). Convert a IntMap into an NEIntMap by adding a key-value pair. Because of this, we know that the map must have at least one element, and so therefore cannot be empty. If key is already present, will overwrite the original value.
See insertMapMin for a version that is constant-time if the new key is strictly smaller than all keys in the original map.
insertMap 4 "c" (Data.IntMap.fromList [(5,"a"), (3,"b")]) == fromList ((3,"b") :| [(4,"c"), (5,"a")])
insertMap 4 "c" Data.IntMap.empty == singleton 4 "c"O(log n). Convert a IntMap into an NEIntMap by adding a key-value pair. Because of this, we know that the map must have at least one element, and so therefore cannot be empty. Uses a combining function with the new value as the first argument if the key is already present.
insertMapWith (++) 4 "c" (Data.IntMap.fromList [(5,"a"), (3,"b")]) == fromList ((3,"b") :| [(4,"c"), (5,"a")])
insertMapWith (++) 5 "c" (Data.IntMap.fromList [(5,"a"), (3,"b")]) == fromList ((3,"b") :| [(5,"ca")])O(log n). Convert a IntMap into an NEIntMap by adding a key-value pair. Because of this, we know that the map must have at least one element, and so therefore cannot be empty. Uses a combining function with the key and new value as the first and second arguments if the key is already present.
let f key new_value old_value = (show key) ++ ":" ++ new_value ++ "|" ++ old_value
insertWithKey f 5 "xxx" (Data.IntMap.fromList [(5,"a"), (3,"b")]) == fromList ((3, "b") :| [(5, "5:xxx|a")])
insertWithKey f 7 "xxx" (Data.IntMap.fromList [(5,"a"), (3,"b")]) == fromList ((3, "b") :| [(5, "a"), (7, "xxx")])
insertWithKey f 5 "xxx" Data.IntMap.empty == singleton 5 "xxx"O(1) Convert a IntMap into an NEIntMap by adding a key-value pair where the key is strictly less than all keys in the input map. The keys in the original map must all be strictly greater than the new key. The precondition is not checked.
insertMapMin 2 "c" (Data.IntMap.fromList [(5,"a"), (3,"b")]) == fromList ((2,"c") :| [(3,"b"), (5,"a")])
valid (insertMapMin 2 "c" (Data.IntMap.fromList [(5,"a"), (3,"b")])) == True
valid (insertMapMin 7 "c" (Data.IntMap.fromList [(5,"a"), (3,"b")])) == False
valid (insertMapMin 3 "c" (Data.IntMap.fromList [(5,"a"), (3,"b")])) == FalseO(log n) Convert a IntMap into an NEIntMap by adding a key-value pair where the key is strictly greater than all keys in the input map. The keys in the original map must all be strictly less than the new key. The precondition is not checked.
At the current moment, this is identical simply insertMap; however, it is left both for consistency and as a placeholder for a future version where optimizations are implemented to allow for a faster implementation.
insertMap 7 "c" (Data.IntMap.fromList [(5,"a"), (3,"b")]) == fromList ((3,"b") :| [(5,"a"), (7,"c")])O(log n). Unsafe version of nonEmptyMap. Coerces a IntMap into an NEIntMap, but is undefined (throws a runtime exception when evaluation is attempted) for an empty IntMap.
Construction
2 declarationsO(n). Build a non-empty map from a non-empty set of keys and a function which for each key computes its value.
fromSet (\k -> replicate k 'a') (Data.Set.NonEmpty.fromList (3 :| [5])) == fromList ((5,"aaaaa") :| [(3,"aaa")])From Unordered Lists
O(n*log n). Build a map from a non-empty list of key/value pairs with a combining function. See also fromAscListWith.
fromListWith (++) ((5,"a") :| [(5,"b"), (3,"b"), (3,"a"), (5,"a")]) == fromList ((3, "ab") :| [(5, "aba")])O(n*log n). Build a map from a non-empty list of key/value pairs with a combining function. See also fromAscListWithKey.
let f k a1 a2 = (show k) ++ a1 ++ a2
fromListWithKey f ((5,"a") :| [(5,"b"), (3,"b"), (3,"a"), (5,"a")]) == fromList ((3, "3ab") :| [(5, "5a5ba")])From Ascending Lists
O(n). Build a map from an ascending non-empty list in linear time. The precondition (input list is ascending) is not checked.
fromAscList ((3,"b") :| [(5,"a")]) == fromList ((3, "b") :| [(5, "a")])
fromAscList ((3,"b") :| [(5,"a"), (5,"b")]) == fromList ((3, "b") :| [(5, "b")])
valid (fromAscList ((3,"b") :| [(5,"a"), (5,"b")])) == True
valid (fromAscList ((5,"a") :| [(3,"b"), (5,"b")])) == FalseO(n). Build a map from an ascending non-empty list in linear time with a combining function for equal keys. /The precondition (input list is ascending) is not checked./
fromAscListWith (++) ((3,"b") :| [(5,"a"), (5,"b")]) == fromList ((3, "b") :| [(5, "ba")])
valid (fromAscListWith (++) ((3,"b") :| [(5,"a"), (5,"b"))]) == True
valid (fromAscListWith (++) ((5,"a") :| [(3,"b"), (5,"b"))]) == FalseO(n). Build a map from an ascending non-empty list in linear time with a combining function for equal keys. /The precondition (input list is ascending) is not checked./
let f k a1 a2 = (show k) ++ ":" ++ a1 ++ a2
fromAscListWithKey f ((3,"b") :| [(5,"a"), (5,"b"), (5,"b")]) == fromList ((3, "b") :| [(5, "5:b5:ba")])
valid (fromAscListWithKey f ((3,"b") :| [(5,"a"), (5,"b"), (5,"b")])) == True
valid (fromAscListWithKey f ((5,"a") :| [(3,"b"), (5,"b"), (5,"b")])) == FalseO(n). Build a map from an ascending non-empty list of distinct elements in linear time. The precondition is not checked.
fromDistinctAscList ((3,"b") :| [(5,"a")]) == fromList ((3, "b") :| [(5, "a")])
valid (fromDistinctAscList ((3,"b") :| [(5,"a")])) == True
valid (fromDistinctAscList ((3,"b") :| [(5,"a"), (5,"b")])) == FalseInsertion
4 declarationsO(log n). Insert a new key and value in the map.
If the key is already present in the map, the associated value is
replaced with the supplied value. insert is equivalent to
insertWith const.
See insertMap for a version where the first argument is a IntMap.
insert 5 'x' (fromList ((5,'a') :| [(3,'b')])) == fromList ((3, 'b') :| [(5, 'x')])
insert 7 'x' (fromList ((5,'a') :| [(3,'b')])) == fromList ((3, 'b') :| [(5, 'a'), (7, 'x')])O(log n). Insert with a function, combining key, new value and old
value. insertWithKey f key value mp will insert the pair (key,
value) into mp if key does not exist in the map. If the key does
exist, the function will insert the pair (key,f key new_value
old_value). Note that the key passed to f is the same key passed to
insertWithKey.
See insertMapWithKey for a version where the first argument is a IntMap.
let f key new_value old_value = (show key) ++ ":" ++ new_value ++ "|" ++ old_value
insertWithKey f 5 "xxx" (fromList ((5,"a") :| [(3,"b")])) == fromList ((3, "b") :| [(5, "5:xxx|a")])
insertWithKey f 7 "xxx" (fromList ((5,"a") :| [(3,"b")])) == fromList ((3, "b") :| [(5, "a"), (7, "xxx")])O(log n). Combines insert operation with old value retrieval. The
expression (insertLookupWithKey f k x map) is a pair where the first
element is equal to (lookup k map) and the second element equal to
(insertWithKey f k x map).
let f key new_value old_value = (show key) ++ ":" ++ new_value ++ "|" ++ old_value
insertLookupWithKey f 5 "xxx" (fromList ((5,"a") :| [(3,"b")])) == (Just "a", fromList ((3, "b") :| [(5, "5:xxx|a")]))
insertLookupWithKey f 7 "xxx" (fromList ((5,"a") :| [(3,"b")])) == (Nothing, fromList ((3, "b") :| [(5, "a"), (7, "xxx")]))This is how to define insertLookup using insertLookupWithKey:
let insertLookup kx x t = insertLookupWithKey (\_ a _ -> a) kx x t
insertLookup 5 "x" (fromList ((5,"a") :| [(3,"b")])) == (Just "a", fromList ((3, "b") :| [(5, "x")]))
insertLookup 7 "x" (fromList ((5,"a") :| [(3,"b")])) == (Nothing, fromList ((3, "b") :| [(5, "a"), (7, "x")]))Deletion/Update
11 declarationsO(log n). Delete a key and its value from the non-empty map. A potentially empty map (IntMap) is returned, since this might delete the last item in the NEIntMap. When the key is not a member of the map, is equivalent to toMap.
delete 5 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 3 "b"
delete 7 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.Singleton [(3, "b"), (5, "a")]O(log n). Delete a key and its value from the non-empty map, returning Nothing if the result would be empty.
This is more efficient than nonEmptyMap . delete k because it avoids
converting the known-minimum representation back through IntMap when the
deleted key is not the minimum.
O(log n). Update a value at a specific key with the result of the provided function. When the key is not a member of the map, the original map is returned.
adjust ("new " ++) 5 (fromList ((5,"a") :| [(3,"b")])) == fromList ((3, "b") :| [(5, "new a")])
adjust ("new " ++) 7 (fromList ((5,"a") :| [(3,"b")])) == fromList ((3, "b") :| [(5, "a")])O(log n). Adjust a value at a specific key. When the key is not a member of the map, the original map is returned.
let f key x = (show key) ++ ":new " ++ x
adjustWithKey f 5 (fromList ((5,"a") :| [(3,"b")])) == fromList ((3, "b") :| [(5, "5:new a")])
adjustWithKey f 7 (fromList ((5,"a") :| [(3,"b")])) == fromList ((3, "b") :| [(5, "a")])O(log n). The expression (update f k map) updates the value x
at k (if it is in the map). If (f x) is Nothing, the element is
deleted. If it is (Just y), the key k is bound to the new value y.
Returns a potentially empty map (IntMap), because we can't know ahead of time if the function returns Nothing and deletes the final item in the NEIntMap.
let f x = if x == "a" then Just "new a" else Nothing
update f 5 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "b"), (5, "new a")]
update f 7 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "b"), (5, "a")]
update f 3 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 5 "a"O(log n). The expression (updateWithKey f k map) updates the
value x at k (if it is in the map). If (f k x) is Nothing,
the element is deleted. If it is (Just y), the key k is bound
to the new value y.
Returns a potentially empty map (IntMap), because we can't know ahead of time if the function returns Nothing and deletes the final item in the NEIntMap.
let f k x = if x == "a" then Just ((show k) ++ ":new a") else Nothing
updateWithKey f 5 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "b"), (5, "5:new a")]
updateWithKey f 7 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "b"), (5, "a")]
updateWithKey f 3 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 5 "a"O(min(n,W)). Lookup and update.
The function returns original value, if it is updated.
This is different behavior than Data.Map.NonEmpty.updateLookupWithKey.
Returns the original key value if the map entry is deleted.
Returns a potentially empty map (IntMap) in the case that we delete the final key of a singleton map.
let f k x = if x == "a" then Just ((show k) ++ ":new a") else Nothing
updateLookupWithKey f 5 (fromList ((5,"a") :| [(3,"b")])) == (Just "5:new a", Data.IntMap.fromList ((3, "b") :| [(5, "5:new a")]))
updateLookupWithKey f 7 (fromList ((5,"a") :| [(3,"b")])) == (Nothing, Data.IntMap.fromList ((3, "b") :| [(5, "a")]))
updateLookupWithKey f 3 (fromList ((5,"a") :| [(3,"b")])) == (Just "b", Data.IntMap.singleton 5 "a")O(log n). The expression (alter f k map) alters the value x at
k, or absence thereof. alter can be used to insert, delete, or
update a value in a IntMap. In short : Data.IntMap.lookup k (alter
f k m) = f (lookup k m).
Returns a potentially empty map (IntMap), because we can't know ahead of time if the function returns Nothing and deletes the final item in the NEIntMap.
See alterF' for a version that disallows deletion, and so therefore can return NEIntMap.
let f _ = Nothing
alter f 7 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "b"), (5, "a")]
alter f 5 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 3 "b"
let f _ = Just "c"
alter f 7 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "b"), (5, "a"), (7, "c")]
alter f 5 (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "b"), (5, "c")]O(log n). The expression (alterF f k map) alters the value x
at k, or absence thereof. alterF can be used to inspect, insert,
delete, or update a value in a IntMap. In short: Data.IntMap.lookup
k <$> alterF f k m = f (lookup k m).
Example:
interactiveAlter :: Int -> NEIntMap Int String -> IO (IntMap Int String)
interactiveAlter k m = alterF f k m where
f Nothing = do
putStrLn $ show k ++
" was not found in the map. Would you like to add it?"
getUserResponse1 :: IO (Maybe String)
f (Just old) = do
putStrLn $ "The key is currently bound to " ++ show old ++
". Would you like to change or delete it?"
getUserResponse2 :: IO (Maybe String)
Like Data.IntMap.alterF for IntMap, alterF can be considered
to be a unifying generalization of lookup and delete; however, as
a constrast, it cannot be used to implement insert, because it must
return a IntMap instead of an NEIntMap (because the function might delete
the final item in the NEIntMap). When used with trivial functors like
Identity and Const, it is often slightly slower than
specialized lookup and delete. However, when the functor is
non-trivial and key comparison is not particularly cheap, it is the
fastest way.
See alterF' for a version that disallows deletion, and so therefore can return NEIntMap and be used to implement insert
Note on rewrite rules:
This module includes GHC rewrite rules to optimize alterF for the Const and Identity functors. In general, these rules improve performance. The sole exception is that when using Identity, deleting a key that is already absent takes longer than it would without the rules. If you expect this to occur a very large fraction of the time, you might consider using a private copy of the Identity type.
Note: Unlike Data.IntMap.alterF for IntMap, alterF is not a flipped
version of the at combinator from Control.Lens.At.
However, it match the shape expected from most functions expecting
lenses, getters, and setters, so can be thought of as a "psuedo-lens",
with virtually the same practical applications as a legitimate lens.
O(log n). Variant of alter that disallows deletion. Allows us to guarantee that the result is also a non-empty IntMap.
O(log n). Variant of alterF that disallows deletion. Allows us to guarantee that the result is also a non-empty IntMap.
Like Data.IntMap.alterF for IntMap, can be used to generalize and unify
lookup and insert. However, because it disallows deletion, it
cannot be used to implement delete.
See alterF for usage information and caveats.
Note: Neither alterF nor alterF' can be considered flipped versions
of the at combinator from Control.Lens.At. However,
this can match the shape expected from most functions expecting lenses,
getters, and setters, so can be thought of as a "psuedo-lens", with
virtually the same practical applications as a legitimate lens.
WARNING: The rewrite rule for Identity exposes an inconsistency in
undefined behavior for Data.IntMap. Data.IntMap.alterF will actually
maintain the original key in the map when used with Identity;
however, Data.IntMap.insertWith will replace the orginal key in the
map. The rewrite rule for alterF' has chosen to be faithful to
Data.IntMap.insertWith, and not Data.IntMap.alterF, for the sake of
a cleaner implementation.
Query
0 declarationsLookup
O(log n). Lookup the value at a key in the map.
The function will return the corresponding value as (Just value),
or Nothing if the key isn't in the map.
An example of using lookup:
import Prelude hiding (lookup)
import Data.Map.NonEmpty
employeeDept = fromList (("John","Sales") :| [("Bob","IT")])
deptCountry = fromList (("IT","USA") :| [("Sales","France")])
countryCurrency = fromList (("USA", "Dollar") :| [("France", "Euro")])
employeeCurrency :: String -> Maybe String
employeeCurrency name = do
dept <- lookup name employeeDept
country <- lookup dept deptCountry
lookup country countryCurrency
main = do
putStrLn $ "John's currency: " ++ (show (employeeCurrency "John"))
putStrLn $ "Pete's currency: " ++ (show (employeeCurrency "Pete"))The output of this program:
John's currency: Just "Euro"
Pete's currency: NothingO(log n). Find the value at a key. Returns Nothing when the element can not be found.
fromList ((5, 'a') :| [(3, 'b')]) !? 1 == NothingfromList ((5, 'a') :| [(3, 'b')]) !? 5 == Just 'a'O(log n). Find the value at a key. Calls error when the element can not be found.
fromList ((5,'a') :| [(3,'b')]) ! 1 Error: element not in the map
fromList ((5,'a') :| [(3,'b')]) ! 5 == 'a'O(log n). The expression (findWithDefault def k map) returns
the value at key k or returns default value def
when the key is not in the map.
findWithDefault 'x' 1 (fromList ((5,'a') :| [(3,'b')])) == 'x'
findWithDefault 'x' 5 (fromList ((5,'a') :| [(3,'b')])) == 'a'O(log n). Is the key a member of the map? See also notMember.
member 5 (fromList ((5,'a') :| [(3,'b')])) == True
member 1 (fromList ((5,'a') :| [(3,'b')])) == FalseO(log n). Is the key not a member of the map? See also member.
notMember 5 (fromList ((5,'a') :| [(3,'b')])) == False
notMember 1 (fromList ((5,'a') :| [(3,'b')])) == TrueO(log n). Find largest key smaller than the given one and return the corresponding (key, value) pair.
lookupLT 3 (fromList ((3,'a') :| [(5,'b')])) == Nothing
lookupLT 4 (fromList ((3,'a') :| [(5,'b')])) == Just (3, 'a')O(log n). Find smallest key greater than the given one and return the corresponding (key, value) pair.
lookupGT 4 (fromList ((3,'a') :| [(5,'b')])) == Just (5, 'b')
lookupGT 5 (fromList ((3,'a') :| [(5,'b')])) == NothingO(log n). Find largest key smaller or equal to the given one and return the corresponding (key, value) pair.
lookupLE 2 (fromList ((3,'a') :| [(5,'b')])) == Nothing
lookupLE 4 (fromList ((3,'a') :| [(5,'b')])) == Just (3, 'a')
lookupLE 5 (fromList ((3,'a') :| [(5,'b')])) == Just (5, 'b')O(log n). Find smallest key greater or equal to the given one and return the corresponding (key, value) pair.
lookupGE 3 (fromList ((3,'a') :| [(5,'b')])) == Just (3, 'a')
lookupGE 4 (fromList ((3,'a') :| [(5,'b')])) == Just (5, 'b')
lookupGE 6 (fromList ((3,'a') :| [(5,'b')])) == NothingSize
Combine
0 declarationsUnion
O(m*log(n/m + 1)), m <= n. Left-biased union of a possibly-empty IntMap and a non-empty map.
O(m*log(n/m + 1)), m <= n. Left-biased union of a non-empty map and a possibly-empty IntMap.
O(m*log(n/m + 1)), m <= n. Union with a combining function.
unionWith (++) (fromList ((5, "a") :| [(3, "b")])) (fromList ((5, "A") :| [(7, "C")])) == fromList ((3, "b") :| [(5, "aA"), (7, "C")])O(m*log(n/m + 1)), m <= n. Union of a possibly-empty IntMap and a non-empty map with a combining function.
O(m*log(n/m + 1)), m <= n. Union of a non-empty map and a possibly-empty IntMap with a combining function.
O(m*log(n/m + 1)), m <= n. Union with a combining function, given the matching key.
let f key left_value right_value = (show key) ++ ":" ++ left_value ++ "|" ++ right_value
unionWithKey f (fromList ((5, "a") :| [(3, "b")])) (fromList ((5, "A") :| [(7, "C")])) == fromList ((3, "b") :| [(5, "5:a|A"), (7, "C")])O(m*log(n/m + 1)), m <= n. Union of a possibly-empty IntMap and a non-empty map with a combining function, given the matching key.
O(m*log(n/m + 1)), m <= n. Union of a non-empty map and a possibly-empty IntMap with a combining function, given the matching key.
The union of a non-empty list of maps, with a combining operation:
(unionsWith f == foldl1 (unionWith f)).
unionsWith (++) (fromList ((5, "a") :| [(3, "b")]) :| [fromList ((5, "A") :| [(7, "C")]), fromList ((5, "A3") :| [(3, "B3")])])
== fromList ((3, "bB3") :| [(5, "aAA3"), (7, "C")])Difference
O(m*log(n/m + 1)), m <= n. Difference of two maps. Return elements of the first map not existing in the second map.
Returns a potentially empty map (IntMap), in case the first map is a subset of the second map.
difference (fromList ((5, "a") :| [(3, "b")])) (fromList ((5, "A") :| [(7, "C")])) == Data.IntMap.singleton 3 "b"Same as difference.
O(n+m). Difference with a combining function.
When two equal keys are
encountered, the combining function is applied to the values of these keys.
If it returns Nothing, the element is discarded (proper set difference). If
it returns (Just y), the element is updated with a new value y.
Returns a potentially empty map (IntMap), in case the first map is a subset of the second map and the function returns Nothing for every pair.
let f al ar = if al == "b" then Just (al ++ ":" ++ ar) else Nothing
differenceWith f (fromList ((5, "a") :| [(3, "b")])) (fromList ((5, "A") :| [(3, "B"), (7, "C")]))
== Data.IntMap.singleton 3 "b:B"O(n+m). Difference with a combining function. When two equal keys are
encountered, the combining function is applied to the key and both values.
If it returns Nothing, the element is discarded (proper set difference). If
it returns (Just y), the element is updated with a new value y.
Returns a potentially empty map (IntMap), in case the first map is a subset of the second map and the function returns Nothing for every pair.
let f k al ar = if al == "b" then Just ((show k) ++ ":" ++ al ++ "|" ++ ar) else Nothing
differenceWithKey f (fromList ((5, "a") :| [(3, "b")])) (fromList ((5, "A") :| [(3, "B"), (10, "C")]))
== Data.IntMap.singleton 3 "3:b|B"Intersection
O(m*log(n/m + 1)), m <= n. Intersection of two maps.
Return data in the first map for the keys existing in both maps.
(intersection m1 m2 == intersectionWith const m1 m2).
Returns a potentially empty map (IntMap), in case the two maps share no keys in common.
intersection (fromList ((5, "a") :| [(3, "b")])) (fromList ((5, "A") :| [(7, "C")])) == Data.IntMap.singleton 5 "a"O(m*log(n/m + 1)), m <= n. Intersection with a combining function.
Returns a potentially empty map (IntMap), in case the two maps share no keys in common.
intersectionWith (++) (fromList ((5, "a") :| [(3, "b")])) (fromList ((5, "A") :| [(7, "C")])) == Data.IntMap.singleton 5 "aA"O(m*log(n/m + 1)), m <= n. Intersection with a combining function.
Returns a potentially empty map (IntMap), in case the two maps share no keys in common.
let f k al ar = (show k) ++ ":" ++ al ++ "|" ++ ar
intersectionWithKey f (fromList ((5, "a") :| [(3, "b")])) (fromList ((5, "A") :| [(7, "C")])) == Data.IntMap.singleton 5 "5:a|A"Traversal
0 declarationsMap
O(n). IntMap a function over all values in the map.
let f key x = (show key) ++ ":" ++ x
mapWithKey f (fromList ((5,"a") :| [(3,"b")])) == fromList ((3, "3:b") :| [(5, "5:a")])O(n). The function mapAccum threads an accumulating argument through the map in ascending order of keys.
let f a b = (a ++ b, b ++ "X")
mapAccum f "Everything: " (fromList ((5,"a") :| [(3,"b")])) == ("Everything: ba", fromList ((3, "bX") :| [(5, "aX")]))O(n). The function mapAccumWithKey threads an accumulating argument through the map in ascending order of keys.
let f a k b = (a ++ " " ++ (show k) ++ "-" ++ b, b ++ "X")
mapAccumWithKey f "Everything:" (fromList ((5,"a") :| [(3,"b")])) == ("Everything: 3-b 5-a", fromList ((3, "bX") :| [(5, "aX")]))O(n). The function mapAccumRWithKey threads an accumulating argument through the map in descending order of keys.
O(n*log n).
mapKeys f s is the map obtained by applying f to each key of s.
The size of the result may be smaller if f maps two or more distinct
keys to the same new key. In this case the value at the greatest of the
original keys is retained.
While the size of the result map may be smaller than the input map, the output map is still guaranteed to be non-empty if the input map is non-empty.
mapKeys (+ 1) (fromList ((5,"a") :| [(3,"b")])) == fromList ((4, "b") :| [(6, "a")])
mapKeys (\ _ -> 1) (fromList ((1,"b") :| [(2,"a"), (3,"d"), (4,"c")])) == singleton 1 "c"
mapKeys (\ _ -> 3) (fromList ((1,"b") :| [(2,"a"), (3,"d"), (4,"c")])) == singleton 3 "c"O(n*log n).
mapKeysWith c f s is the map obtained by applying f to each key of s.
The size of the result may be smaller if f maps two or more distinct
keys to the same new key. In this case the associated values will be
combined using c. The value at the greater of the two original keys
is used as the first argument to c.
While the size of the result map may be smaller than the input map, the output map is still guaranteed to be non-empty if the input map is non-empty.
mapKeysWith (++) (\ _ -> 1) (fromList ((1,"b") :| [(2,"a"), (3,"d"), (4,"c")])) == singleton 1 "cdab"
mapKeysWith (++) (\ _ -> 3) (fromList ((1,"b") :| [(2,"a"), (3,"d"), (4,"c")])) == singleton 3 "cdab"O(n).
mapKeysMonotonic f s == mapKeys f s, but works only when f
is strictly monotonic.
That is, for any values x and y, if x < y then f x < f y.
The precondition is not checked.
Semi-formally, we have:
and [x < y ==> f x < f y | x <- ls, y <- ls]
==> mapKeysMonotonic f s == mapKeys f s
where ls = keys sThis means that f maps distinct original keys to distinct resulting keys.
This function has better performance than mapKeys.
While the size of the result map may be smaller than the input map, the output map is still guaranteed to be non-empty if the input map is non-empty.
mapKeysMonotonic (\ k -> k * 2) (fromList ((5,"a") :| [(3,"b")])) == fromList ((6, "b") :| [(10, "a")])
valid (mapKeysMonotonic (\ k -> k * 2) (fromList ((5,"a") :| [(3,"b")]))) == True
valid (mapKeysMonotonic (\ _ -> 1) (fromList ((5,"a") :| [(3,"b")]))) == FalseFolds
7 declarationsO(n). Fold the keys and values in the map using the given right-associative
binary operator, such that
foldrWithKey f z == .Prelude.foldr (uncurry f) z . toAscList
For example,
keysList map = foldrWithKey (\k x ks -> k:ks) [] mapO(n). Fold the keys and values in the map using the given left-associative
binary operator, such that
foldlWithKey f z == .Prelude.foldl (\z' (kx, x) -> f z' kx x) z . toAscList
For example,
keysList = reverse . foldlWithKey (\ks k x -> k:ks) []Strict folds
O(n). A strict version of foldr1. Each application of the operator is evaluated before using the result in the next application. This function is strict in the starting value.
O(n). A strict version of foldl1. Each application of the operator is evaluated before using the result in the next application. This function is strict in the starting value.
O(n). A strict version of foldrWithKey. Each application of the operator is evaluated before using the result in the next application. This function is strict in the starting value.
O(n). A strict version of foldlWithKey. Each application of the operator is evaluated before using the result in the next application. This function is strict in the starting value.
Conversion
4 declarationsO(n). Return all keys of the map in ascending order.
keys (fromList ((5,"a") :| [(3,"b")])) == (3 :| [5])O(n). An alias for toAscList. Return all key/value pairs in the map in ascending key order.
assocs (fromList ((5,"a") :| [(3,"b")])) == ((3,"b") :| [(5,"a")])O(n). The non-empty set of all keys of the map.
keysSet (fromList ((5,"a") :| [(3,"b")])) == Data.Set.NonEmpty.fromList (3 :| [5])Lists
Ordered lists
O(n). Convert the map to a list of key/value pairs where the keys are in ascending order.
toAscList (fromList ((5,"a") :| [(3,"b")])) == ((3,"b") :| [(5,"a")])O(n). Convert the map to a list of key/value pairs where the keys are in descending order.
toDescList (fromList ((5,"a") :| [(3,"b")])) == ((5,"a") :| [(3,"b")])Filter
13 declarationsO(n). Filter all values that satisfy the predicate.
Returns a potentially empty map (IntMap), because we could potentailly filter out all items in the original NEIntMap.
filter (> "a") (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 3 "b"
filter (> "x") (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.empty
filter (< "a") (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.emptyO(m*log(n/m + 1)), m <= n. Restrict an NEIntMap to only those keys found in a Set.
m `restrictKeys` s = filterWithKey (k _ -> k `Set.member` s) m
m `restrictKeys` s = m `intersection` fromSet (const ()) s
O(m*log(n/m + 1)), m <= n. Remove all keys in a Set from an NEIntMap.
m `withoutKeys` s = filterWithKey (k _ -> k `Set.notMember` s) m
m `withoutKeys` s = m `difference` fromSet (const ()) s
O(n). Partition the map according to a predicate.
Returns a These with potentially two non-empty maps:
This n1means that the predicate was true for all items.That n2means that the predicate was false for all items.These n1 n2givesn1(all of the items that were true for the predicate) andn2(all of the items that were false for the predicate).
See also split.
partition (> "a") (fromList ((5,"a") :| [(3,"b")])) == These (singleton 3 "b") (singleton 5 "a")
partition (< "x") (fromList ((5,"a") :| [(3,"b")])) == This (fromList ((3, "b") :| [(5, "a")]))
partition (> "x") (fromList ((5,"a") :| [(3,"b")])) == That (fromList ((3, "b") :| [(5, "a")]))O(n). Partition the map according to a predicate.
Returns a These with potentially two non-empty maps:
This n1means that the predicate was true for all items, returning the original map.That n2means that the predicate was false for all items, returning the original map.These n1 n2givesn1(all of the items that were true for the predicate) andn2(all of the items that were false for the predicate).
See also split.
partitionWithKey (\ k _ -> k > 3) (fromList ((5,"a") :| [(3,"b")])) == These (singleton 5 "a") (singleton 3 "b")
partitionWithKey (\ k _ -> k < 7) (fromList ((5,"a") :| [(3,"b")])) == This (fromList ((3, "b") :| [(5, "a")]))
partitionWithKey (\ k _ -> k > 7) (fromList ((5,"a") :| [(3,"b")])) == That (fromList ((3, "b") :| [(5, "a")]))O(n). Map values and collect the Just results.
Returns a potentially empty map (IntMap), because the function could potentially return Nothing on all items in the NEIntMap.
let f x = if x == "a" then Just "new a" else Nothing
mapMaybe f (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 5 "new a"O(n). Map keys/values and collect the Just results.
Returns a potentially empty map (IntMap), because the function could potentially return Nothing on all items in the NEIntMap.
let f k _ = if k < 5 then Just ("key : " ++ (show k)) else Nothing
mapMaybeWithKey f (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 3 "key : 3"O(n). Map values and separate the Left and Right results.
Returns a These with potentially two non-empty maps:
These n1 n2givesn1(the map where the results were Left) andn2(the map where the results were Right)
let f a = if a < "c" then Left a else Right a
mapEither f (fromList ((5,"a") :| [(3,"b"), (1,"x"), (7,"z")]))
== These (fromList ((3,"b") :| [(5,"a")])) (fromList ((1,"x") :| [(7,"z")]))
mapEither (\ a -> Right a) (fromList ((5,"a") :| [(3,"b"), (1,"x"), (7,"z")]))
== That (fromList ((5,"a") :| [(3,"b"), (1,"x"), (7,"z")]))O(n). Map keys/values and separate the Left and Right results.
Returns a These with potentially two non-empty maps:
These n1 n2givesn1(the map where the results were Left) andn2(the map where the results were Right)
let f k a = if k < 5 then Left (k * 2) else Right (a ++ a)
mapEitherWithKey f (fromList ((5,"a") :| [(3,"b"), (1,"x"), (7,"z")]))
== These (fromList ((1,2) :| [(3,6)])) (fromList ((5,"aa") :| [(7,"zz")]))
mapEitherWithKey (\_ a -> Right a) (fromList ((5,"a") :| [(3,"b"), (1,"x"), (7,"z")]))
== That (fromList ((1,"x") :| [(3,"b"), (5,"a"), (7,"z")]))O(log n). The expression (split k map) is potentially a These
containing up to two NEIntMaps based on splitting the map into maps
containing items before and after the given key k. It will never
return a map that contains k itself.
Nothing means that
kwas the only key in the the original map, and so there are no items before or after it.Just (This n1)meanskwas larger than or equal to all items in the map, andn1is the entire original map (minusk, if it was present)Just (That n2)meanskwas smaller than or equal to all items in the map, andn2is the entire original map (minusk, if it was present)Just (These n1 n2)givesn1(the map of all keys from the original map less thank) andn2(the map of all keys from the original map greater thank)
split 2 (fromList ((5,"a") :| [(3,"b")])) == Just (That (fromList ((3,"b") :| [(5,"a")])) )
split 3 (fromList ((5,"a") :| [(3,"b")])) == Just (That (singleton 5 "a") )
split 4 (fromList ((5,"a") :| [(3,"b")])) == Just (These (singleton 3 "b") (singleton 5 "a"))
split 5 (fromList ((5,"a") :| [(3,"b")])) == Just (This (singleton 3 "b") )
split 6 (fromList ((5,"a") :| [(3,"b")])) == Just (This (fromList ((3,"b") :| [(5,"a")])) )
split 5 (singleton 5 "a") == NothingO(log n). The expression (splitLookup k map) splits a map just
like split but also returns lookup k map, as the first field in
the These:
splitLookup 2 (fromList ((5,"a") :| [(3,"b")])) == That (That (fromList ((3,"b") :| [(5,"a")])))
splitLookup 3 (fromList ((5,"a") :| [(3,"b")])) == These "b" (That (singleton 5 "a"))
splitLookup 4 (fromList ((5,"a") :| [(3,"b")])) == That (These (singleton 3 "b") (singleton 5 "a"))
splitLookup 5 (fromList ((5,"a") :| [(3,"b")])) == These "a" (This (singleton 3 "b"))
splitLookup 6 (fromList ((5,"a") :| [(3,"b")])) == That (This (fromList ((3,"b") :| [(5,"a")])))
splitLookup 5 (singleton 5 "a") == This "a"O(1). Decompose a map into pieces based on the structure of the underlying tree. This function is useful for consuming a map in parallel.
No guarantee is made as to the sizes of the pieces; an internal, but deterministic process determines this. However, it is guaranteed that the pieces returned will be in ascending order (all elements in the first submap less than all elements in the second, and so on).
Note that the current implementation does not return more than four submaps, but you should not depend on this behaviour because it can change in the future without notice.
Submap
4 declarationsO(m*log(n/m + 1)), m <= n.
This function is defined as (isSubmapOf = isSubmapOfBy (==)).
O(m*log(n/m + 1)), m <= n.
The expression (isSubmapOfBy f t1 t2) returns True if
all keys in t1 are in tree t2, and when f returns True when
applied to their respective values. For example, the following
expressions are all True:
isSubmapOfBy (==) (singleton 'a' 1) (fromList (('a',1) :| [('b',2)]))
isSubmapOfBy (<=) (singleton 'a' 1) (fromList (('a',1) :| [('b',2)]))
isSubmapOfBy (==) (fromList (('a',1) :| [('b',2)])) (fromList (('a',1) :| [('b',2)]))But the following are all False:
isSubmapOfBy (==) (singleton 'a' 2) (fromList (('a',1) :| [('b',2)]))
isSubmapOfBy (<) (singleton 'a' 1) (fromList (('a',1) :| [('b',2)]))
isSubmapOfBy (==) (fromList (('a',1) :| [('b',2)])) (singleton 'a' 1)O(m*log(n/m + 1)), m <= n. Is this a proper submap? (ie. a submap
but not equal). Defined as (isProperSubmapOf = isProperSubmapOfBy
(==)).
O(m*log(n/m + 1)), m <= n. Is this a proper submap? (ie. a submap
but not equal). The expression (isProperSubmapOfBy f m1 m2) returns
True when m1 and m2 are not equal, all keys in m1 are in m2,
and when f returns True when applied to their respective values. For
example, the following expressions are all True:
isProperSubmapOfBy (==) (singleton 1 1) (fromList ((1,1) :| [(2,2)]))
isProperSubmapOfBy (<=) (singleton 1 1) (fromList ((1,1) :| [(2,2)]))But the following are all False:
isProperSubmapOfBy (==) (fromList ((1,1) :| [(2,2)])) (fromList ((1,1) :| [(2,2)]))
isProperSubmapOfBy (==) (fromList ((1,1) :| [(2,2)])) (singleton 1 1))
isProperSubmapOfBy (<) (singleton 1 1) (fromList ((1,1) :| [(2,2)]))Min/Max
16 declarationsO(1). The minimal key of the map. Note that this is total, making
Data.IntMap.lookupMin obsolete. It is constant-time, so has better
asymptotics than Data.IntMap.lookupMin and Data.IntMap.findMin, as well.
findMin (fromList ((5,"a") :| [(3,"b")])) == (3,"b")O(log n). The maximal key of the map. Note that this is total, making
Data.IntMap.lookupMin obsolete.
findMax (fromList ((5,"a") :| [(3,"b")])) == (5,"a")O(1). Delete the minimal key. Returns a potentially empty map
(IntMap), because we might end up deleting the final key in a singleton
map. It is constant-time, so has better asymptotics than
Data.IntMap.deleteMin.
deleteMin (fromList ((5,"a") :| [(3,"b"), (7,"c")])) == Data.IntMap.fromList [(5,"a"), (7,"c")]
deleteMin (singleton 5 "a") == Data.IntMap.emptyO(log n). Delete the maximal key. Returns a potentially empty map (IntMap), because we might end up deleting the final key in a singleton map.
deleteMax (fromList ((5,"a") :| [(3,"b"), (7,"c")])) == Data.IntMap.fromList [(3,"b"), (5,"a")]
deleteMax (singleton 5 "a") == Data.IntMap.emptyO(1). Delete and find the minimal key-value pair. It is
constant-time, so has better asymptotics that Data.IntMap.minView for
IntMap.
Note that unlike Data.IntMap.deleteFindMin for IntMap, this cannot ever
fail, and so is a total function. However, the result IntMap is
potentially empty, since the original map might have contained just
a single item.
deleteFindMin (fromList ((5,"a") :| [(3,"b"), (10,"c")])) == ((3,"b"), Data.IntMap.fromList [(5,"a"), (10,"c")])O(log n). Delete and find the minimal key-value pair.
Note that unlike Data.IntMap.deleteFindMax for IntMap, this cannot ever
fail, and so is a total function. However, the result IntMap is
potentially empty, since the original map might have contained just
a single item.
deleteFindMax (fromList ((5,"a") :| [(3,"b"), (10,"c")])) == ((10,"c"), Data.IntMap.fromList [(3,"b"), (5,"a")])O(1) if delete, O(log n) otherwise. Update the value at the minimal key. Returns a potentially empty map (IntMap), because we might end up deleting the final key in the map if the function returns Nothing. See adjustMin for a version that can guaruntee that we return a non-empty map.
updateMin (\ a -> Just ("X" ++ a)) (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "Xb"), (5, "a")]
updateMin (\ _ -> Nothing) (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 5 "a"O(log n). Update the value at the maximal key. Returns a potentially empty map (IntMap), because we might end up deleting the final key in the map if the function returns Nothing. See adjustMax for a version that can guarantee that we return a non-empty map.
updateMax (\ a -> Just ("X" ++ a)) (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3, "b"), (5, "Xa")]
updateMax (\ _ -> Nothing) (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 3 "b"O(1). A version of updateMin that disallows deletion, allowing us to guarantee that the result is also non-empty.
O(log n). A version of updateMax that disallows deletion, allowing us to guarantee that the result is also non-empty.
O(1) if delete, O(log n) otherwise. Update the value at the minimal key. Returns a potentially empty map (IntMap), because we might end up deleting the final key in the map if the function returns Nothing. See adjustMinWithKey for a version that guaruntees a non-empty map.
updateMinWithKey (\ k a -> Just ((show k) ++ ":" ++ a)) (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3,"3:b"), (5,"a")]
updateMinWithKey (\ _ _ -> Nothing) (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 5 "a"O(log n). Update the value at the maximal key. Returns a potentially empty map (IntMap), because we might end up deleting the final key in the map if the function returns Nothing. See adjustMaxWithKey for a version that guaruntees a non-empty map.
updateMinWithKey (\ k a -> Just ((show k) ++ ":" ++ a)) (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.fromList [(3,"3:b"), (5,"a")]
updateMinWithKey (\ _ _ -> Nothing) (fromList ((5,"a") :| [(3,"b")])) == Data.IntMap.singleton 5 "a"O(1). A version of adjustMaxWithKey that disallows deletion, allowing us to guarantee that the result is also non-empty. Note that it also is able to have better asymptotics than updateMinWithKey in general.
O(log n). A version of updateMaxWithKey that disallows deletion, allowing us to guarantee that the result is also non-empty.
O(1). Retrieves the value associated with minimal key of the
map, and the map stripped of that element. It is constant-time, so has
better asymptotics than Data.IntMap.minView for IntMap.
Note that unlike Data.IntMap.minView for IntMap, this cannot ever fail,
so doesn't need to return in a Maybe. However, the result IntMap is
potentially empty, since the original map might have contained just
a single item.
minView (fromList ((5,"a") :| [(3,"b")])) == ("b", Data.IntMap.singleton 5 "a")O(log n). Retrieves the value associated with maximal key of the map, and the map stripped of that element.
Note that unlike Data.IntMap.maxView from IntMap, this cannot ever fail,
so doesn't need to return in a Maybe. However, the result IntMap is
potentially empty, since the original map might have contained just
a single item.
maxView (fromList ((5,"a") :| [(3,"b")])) == ("a", Data.IntMap.singleton 3 "b")