This is an alternative site for discovering Elm packages.
You may be looking for the
official Elm package site
instead.

Convenience functions for working with List

last : List a -> Maybe a

Extract the last element of a list.

```
last [1,2,3] == Just 3
last [] == Nothing
```

init : List a -> Maybe (List a)

Return all elements of the list except the last one.

```
init [1,2,3] == Just [1,2]
init [] == Nothing
```

getAt : List a -> Int -> Maybe a

Returns `Just`

the element at the given index in the list,
or `Nothing`

if the list is not long enough.

uncons : List a -> Maybe (a, List a)

Decompose a list into its head and tail. If the list is empty, return `Nothing`

. Otherwise, return `Just (x, xs)`

, where `x`

is head and `xs`

is tail.

```
uncons [1,2,3] == Just (1, [2,3])
uncons [] = Nothing
```

maximumBy : (a -> comparable) -> List a -> Maybe a

Find the first maximum element in a list using a comparable transformation

minimumBy : (a -> comparable) -> List a -> Maybe a

Find the first minimum element in a list using a comparable transformation

andMap : List (a -> b) -> List a -> List b

Map functions taking multiple arguments over multiple lists. Each list should be of the same length.

```
( (\a b c -> a + b * c)
`map` [1,2,3]
`andMap` [4,5,6]
`andMap` [2,1,1]
) == [9,7,9]
```

andThen : List a -> (a -> List b) -> List b

Equivalent to `concatMap`

with arguments reversed. Ideal to use as an infix function, chaining together functions that return List. For example, suppose you want to have a cartesian product of [1,2] and [3,4]:

```
[1,2] `andThen` \x ->
[3,4] `andThen` \y ->
[(x,y)]
```

will give back the list:

```
[(1,3),(1,4),(2,3),(2,4)]
```

Now suppose we want to have a cartesian product between the first list and the second list and its doubles:

```
[1,2] `andThen` \x ->
[3,4] `andThen` \y ->
[y,y*2] `andThen` \z ->
[(x,z)]
```

will give back the list:

```
[(1,3),(1,6),(1,4),(1,8),(2,3),(2,6),(2,4),(2,8)]
```

Advanced functional programmers will recognize this as the implementation of bind operator (>>=) for lists from the `Monad`

typeclass.

takeWhile : (a -> Bool) -> List a -> List a

Take elements in order as long as the predicate evaluates to `True`

dropWhile : (a -> Bool) -> List a -> List a

Drop elements in order as long as the predicate evaluates to `True`

dropDuplicates : List comparable -> List comparable

Drop *all* duplicate elements from the list

find : (a -> Bool) -> List a -> Maybe a

Find the first element that satisfies a predicate and return Just that element. If none match, return Nothing.

```
find (\num -> num > 5) [2, 4, 6, 8] == Just 6
```

replaceIf : (a -> Bool) -> a -> List a -> List a

Replace all values that satisfy a predicate with a replacement value.

singleton : a -> List a

Convert a value to a list containing one value.

```
singleton 3 == [3]
```

removeWhen : (a -> Bool) -> List a -> List a

Take a predicate and a list, and return a list that contains elements which fails to satisfy the predicate.
This is equivalent to `List.filter (not << predicate) list`

.

```
removeWhen isEven [1,2,3,4] == [1,3]
```

intercalate : List a -> List (List a) -> List a

Take a list and a list of lists, insert that list between every list in the list of lists, concatenate the result. `intercalate xs xss`

is equivalent to `concat (intersperse xs xss)`

.

```
intercalate [0,0] [[1,2],[3,4],[5,6]] == [1,2,0,0,3,4,0,0,5,6]
```

transpose : List (List a) -> List (List a)

Transpose rows and columns of the list of lists.

```
transpose [[1,2,3],[4,5,6]] == [[1,4],[2,5],[3,6]]
```

If some rows are shorter than the following rows, their elements are skipped:

```
transpose [[10,11],[20],[],[30,31,32]] == [[10,20,30],[11,31],[32]]
```

subsequences : List a -> List (List a)

Return the list of all subsequences of a list.

```
subsequences [1,2,3] == [[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
```

permutations : List a -> List (List a)

Return the list of of all permutations of a list. The result is in lexicographic order.

```
permutations [1,2,3] == [[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
```

interweave : List a -> List a -> List a

Return a list that contains elements from the two provided, in alternate order. If one list runs out of items, append the items from the remaining list.

```
interweave [1,3] [2,4] == [1,2,3,4]
interweave [1,3,5,7] [2,4] == [1,2,3,4,5,7]
```

unique : List comparable -> List comparable

Remove all duplicates from a list and return a list of distinct elements.

foldl1 : (a -> a -> a) -> List a -> Maybe a

Variant of `foldl`

that has no starting value argument and treats the head of the list as its starting value. If the list is empty, return `Nothing`

.

```
foldl1 max [1,2,3,2,1] == Just 3
foldl1 max [] == Nothing
foldl1 (-) [1,2,3] == Just -4
```

foldr1 : (a -> a -> a) -> List a -> Maybe a

Variant of `foldr`

that has no starting value argument and treats the last element of the list as its starting value. If the list is empty, return `Nothing`

.

```
foldr1 min [1,2,3,2,1] == Just 1
foldr1 min [] == Nothing
foldr1 (-) [1,2,3] == Just 2
```

scanl1 : (a -> a -> a) -> List a -> List a

`scanl1`

is a variant of `scanl`

that has no starting value argument.

Compare:

```
List.scanl (+) 0 [1,2,3] == [0,1,3,6]
scanl1 (+) [1,2,3] == [1,3,6]
List.scanl (-) 0 [1,2,3] == [0,1,1,2]
scanl1 (-) [1,2,3] == [1,1,2]
List.scanl (flip (-)) 0 [1,2,3] == [0,-1,-3,-6]
scanl1 (flip (-)) [1,2,3] == [1,-1,4]
```

scanr : (a -> b -> b) -> b -> List a -> List b

`scanr`

is a right-to-left dual of `scanl`

. Note that:

```
head (scanr f z xs) == foldr f z xs
```

Examples:

```
scanr (+) 0 [1,2,3] == [6,5,3,0]
scanr (-) 0 [1,2,3] == [2,-1,3,0]
```

scanr1 : (a -> a -> a) -> List a -> List a

`scanr1`

is a variant of `scanr`

that has no starting value argument.

```
scanr1 (+) [1,2,3] == [6,5,3]
scanr1 (-) [1,2,3] == [2,-1,3]
scanr1 (flip (-)) [1,2,3] == [0,1,3]
```

unfoldr : (b -> Maybe (a, b)) -> b -> List a

The `unfoldr`

function is "dual" to `foldr`

. `foldr`

reduces a list to a summary value, `unfoldr`

builds a list from a seed. The function takes a function and a starting element. It applies the function to the element. If the result is `Just (a, b)`

, `a`

is accumulated and the function is applied to `b`

. If the result is `Nothing`

, the list accumulated so far is returned.

```
unfoldr (\b -> if b == 0 then Nothing else Just (b, b-1)) 5 == [5,4,3,2,1]
```

iterate : (a -> Maybe a) -> a -> List a

Returns a list of repeated applications of `f`

.

If `f`

returns `Nothing`

the iteration will stop. If it returns `Just y`

then `y`

will be added to the list and the iteration will continue with `f y`

.
nextYear : Int -> Maybe Int
nextYear year =
if year >= 2030 then
Nothing
else
Just (year + 1)
-- Will evaluate to [2010, 2011, ..., 2030]
iterate nextYear 2010

splitAt : Int -> List a -> (List a, List a)

Take a number and a list, return a tuple of lists, where first part is prefix of the list of length equal the number, and second part is the remainder of the list. `splitAt n xs`

is equivalent to `(take n xs, drop n xs)`

.

```
splitAt 3 [1,2,3,4,5] == ([1,2,3],[4,5])
splitAt 1 [1,2,3] == ([1],[2,3])
splitAt 3 [1,2,3] == ([1,2,3],[])
splitAt 4 [1,2,3] == ([1,2,3],[])
splitAt 0 [1,2,3] == ([],[1,2,3])
splitAt (-1) [1,2,3] == ([],[1,2,3])
```

takeWhileEnd : (a -> Bool) -> List a -> List a

Take elements from the end, while predicate still holds.

```
takeWhileEnd ((<)5) [1..10] == [6,7,8,9,10]
```

dropWhileEnd : (a -> Bool) -> List a -> List a

Drop elements from the end, while predicate still holds.

```
dropWhileEnd ((<)5) [1..10] == [1,2,3,4,5]
```

span : (a -> Bool) -> List a -> (List a, List a)

Take a predicate and a list, return a tuple. The first part of the tuple is longest prefix of that list, for each element of which the predicate holds. The second part of the tuple is the remainder of the list. `span p xs`

is equivalent to `(takeWhile p xs, dropWhile p xs)`

.

```
span (< 3) [1,2,3,4,1,2,3,4] == ([1,2],[3,4,1,2,3,4])
span (< 5) [1,2,3] == ([1,2,3],[])
span (< 0) [1,2,3] == ([],[1,2,3])
```

break : (a -> Bool) -> List a -> (List a, List a)

Take a predicate and a list, return a tuple. The first part of the tuple is longest prefix of that list, for each element of which the predicate *does not* hold. The second part of the tuple is the remainder of the list. `span p xs`

is equivalent to `(dropWhile p xs, takeWhile p xs)`

.

```
break (> 3) [1,2,3,4,1,2,3,4] == ([1,2,3],[4,1,2,3,4])
break (< 5) [1,2,3] == ([],[1,2,3])
break (> 5) [1,2,3] == ([1,2,3],[])
```

stripPrefix : List a -> List a -> Maybe (List a)

Drop the given prefix from the list. If the list doesn't start with that prefix, return `Nothing`

.

```
stripPrefix [1,2] [1,2,3,4] == Just [3,4]
stripPrefix [1,2,3] [1,2,3,4,5] == Just [4,5]
stripPrefix [1,2,3] [1,2,3] == Just []
stripPrefix [1,2,3] [1,2] == Nothing
stripPrefix [3,2,1] [1,2,3,4,5] == Nothing
```

group : List a -> List (List a)

Group similar elements together. `group`

is equivalent to `groupBy (==)`

.

```
group [1,2,2,3,3,3,2,2,1] == [[1],[2,2],[3,3,3],[2,2],[1]]
```

groupBy : (a -> a -> Bool) -> List a -> List (List a)

Group elements together, using a custom equality test.

```
groupBy (\x y -> fst x == fst y) [(0,'a'),(0,'b'),(1,'c'),(1,'d')] == [[(0,'a'),(0,'b')],[(1,'c'),(1,'d')]]
```

The equality test should be an equivalent relationship, i.e. it should have the properties of reflexivity, symmetry, and transitivity. For non-equivalent relations it gives non-intuitive behavior:

```
groupBy (<) [1,2,3,2,4,1,3,2,1] == [[1,2,3,2,4],[1,3,2],[1]]
```

For grouping elements with a comparison test, which must only hold the property of transitivity, see `groupByTransitive`

.

groupByTransitive : (a -> a -> Bool) -> List a -> List (List a)

Group elements together, using a custom comparison test. Start a new group each time the comparison test doesn't hold for two adjacent elements.

```
groupByTransitive (<) [1,2,3,2,4,1,3,2,1] == [[1,2,3],[2,4],[1,3],[2],[1]]
```

inits : List a -> List (List a)

Return all initial segments of a list, from shortest to longest, empty list first, the list itself last.

```
inits [1,2,3] == [[],[1],[1,2],[1,2,3]]
```

tails : List a -> List (List a)

Return all final segments of a list, from longest to shortest, the list itself first, empty list last.

```
tails [1,2,3] == [[1,2,3],[2,3],[3],[]]
```

select : List a -> List (a, List a)

Return all combinations in the form of (element, rest of the list). Read Haskell Libraries proposal for further ideas on how to use this function.

```
select [1,2,3,4] == [(1,[2,3,4]),(2,[1,3,4]),(3,[1,2,4]),(4,[1,2,3])]
```

selectSplit : List a -> List (List a, a, List a)

Return all combinations in the form of (elements before, element, elements after).

```
selectSplit [1,2,3] == [([],1,[2,3]),([1],2,[3]),([1,2],3,[])]
```

isPrefixOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is the prefix of the second list.

isSuffixOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is the suffix of the second list.

isInfixOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is an infix of the second list.

isSubsequenceOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is a subsequence of the second list.

isPermutationOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is a permutation of the second list.

notMember : a -> List a -> Bool

Negation of `member`

.

```
1 `notMember` [1,2,3] == False
4 `notMember` [1,2,3] == True
```

find : (a -> Bool) -> List a -> Maybe a

Find the first element that satisfies a predicate and return Just that element. If none match, return Nothing.

```
find (\num -> num > 5) [2, 4, 6, 8] == Just 6
```

elemIndex : a -> List a -> Maybe Int

Return the index of the first occurrence of the element. Otherwise, return `Nothing`

. Indexing starts from 0.

```
elemIndex 1 [1,2,3] == Just 0
elemIndex 4 [1,2,3] == Nothing
elemIndex 1 [1,2,1] == Just 0
```

elemIndices : a -> List a -> List Int

Return all indices of occurrences of the element. If element is not found, return empty list. Indexing starts from 0.

```
elemIndices 1 [1,2,3] == [0]
elemIndices 4 [1,2,3] == []
elemIndices 1 [1,2,1] == [0,2]
```

findIndex : (a -> Bool) -> List a -> Maybe Int

Take a predicate and a list, return the index of the first element that satisfies the predicate. Otherwise, return `Nothing`

. Indexing starts from 0.

```
findIndex isEven [1,2,3] == Just 1
findIndex isEven [1,3,5] == Nothing
findIndex isEven [1,2,4] == Just 1
```

findIndices : (a -> Bool) -> List a -> List Int

Take a predicate and a list, return indices of all elements satisfying the predicate. Otherwise, return empty list. Indexing starts from 0.

```
findIndices isEven [1,2,3] == [1]
findIndices isEven [1,3,5] == []
findIndices isEven [1,2,4] == [1,2]
```

zip : List a -> List b -> List (a,b)

Take two lists and returns a list of corresponding pairs

zip3 : List a -> List b -> List c -> List (a,b,c)

Take three lists and returns a list of triples

zip4 : List a -> List b -> List c -> List d -> List (a,b,c,d)

Take four lists and returns a list of quadruples

zip5 : List a -> List b -> List c -> List d -> List e -> List (a,b,c,d,e)

Take five lists and returns a list of quintuples

lift2 : (a -> b -> c) -> List a -> List b -> List c

Map functions taking multiple arguments over multiple lists, regardless of list length. All possible combinations will be explored.

lift2 (+) [1,2,3] [4,5] == [5,6,6,7,7,8]

lift3 : (a -> b -> c -> d) -> List a -> List b -> List c -> List d

lift4 : (a -> b -> c -> d -> e) -> List a -> List b -> List c -> List d -> List e

```
module List.Extra
( last
, init
, getAt, (!!)
, uncons
, minimumBy
, maximumBy
, andMap, andThen
, takeWhile
, dropWhile
, dropDuplicates
, replaceIf
, singleton
, removeWhen
, iterate
, intercalate, transpose, subsequences, permutations, interweave, unique
, foldl1, foldr1
, scanl1, scanr, scanr1, unfoldr
, splitAt, takeWhileEnd, dropWhileEnd, span, break, stripPrefix
, group, groupBy, groupByTransitive, inits, tails, select, selectSplit
, isPrefixOf, isSuffixOf, isInfixOf, isSubsequenceOf, isPermutationOf
, notMember, find
, elemIndex, elemIndices, findIndex, findIndices
, zip
, zip3
, zip4
, zip5
, lift2
, lift3
, lift4
) where
{-| Convenience functions for working with List
# Basics
@docs last, init, getAt, (!!), uncons, maximumBy, minimumBy, andMap, andThen, takeWhile, dropWhile, dropDuplicates, find, replaceIf, singleton, removeWhen
# List transformations
@docs intercalate, transpose, subsequences, permutations, interweave, unique
# Folds
@docs foldl1, foldr1
# Building lists
@docs scanl1, scanr, scanr1, unfoldr, iterate
# Sublists
@docs splitAt, takeWhileEnd, dropWhileEnd, span, break, stripPrefix, group, groupBy, groupByTransitive, inits, tails, select, selectSplit
# Predicates
@docs isPrefixOf, isSuffixOf, isInfixOf, isSubsequenceOf, isPermutationOf
# Searching
@docs notMember, find, elemIndex, elemIndices, findIndex, findIndices
# Zipping
@docs zip, zip3, zip4, zip5
# Lift functions onto multiple lists of arguments
@docs lift2, lift3, lift4
-}
import List exposing (..)
import Set exposing (Set)
{-| Extract the last element of a list.
last [1,2,3] == Just 3
last [] == Nothing
-}
last : List a -> Maybe a
last = foldl1 (flip always)
{-| Return all elements of the list except the last one.
init [1,2,3] == Just [1,2]
init [] == Nothing
-}
init : List a -> Maybe (List a)
init =
let
maybe : b -> (a -> b) -> Maybe a -> b
maybe d f = Maybe.withDefault d << Maybe.map f
in
foldr ((<<) Just << maybe [] << (::)) Nothing
{-| Returns `Just` the element at the given index in the list,
or `Nothing` if the list is not long enough.
-}
getAt : List a -> Int -> Maybe a
getAt xs idx = List.head <| List.drop idx xs
{-| Alias for getAt
-}
(!!) : List a -> Int -> Maybe a
(!!) = getAt
{-| Returns a list of repeated applications of `f`.
If `f` returns `Nothing` the iteration will stop. If it returns `Just y` then `y` will be added to the list and the iteration will continue with `f y`.
nextYear : Int -> Maybe Int
nextYear year =
if year >= 2030 then
Nothing
else
Just (year + 1)
-- Will evaluate to [2010, 2011, ..., 2030]
iterate nextYear 2010
-}
iterate : (a -> Maybe a) -> a -> List a
iterate f x =
case f x of
Just x' -> x :: iterate f x'
Nothing -> [x]
{-| Decompose a list into its head and tail. If the list is empty, return `Nothing`. Otherwise, return `Just (x, xs)`, where `x` is head and `xs` is tail.
uncons [1,2,3] == Just (1, [2,3])
uncons [] = Nothing
-}
uncons : List a -> Maybe (a, List a)
uncons xs =
case xs of
[] -> Nothing
(x::xs) -> Just (x,xs)
{-| Find the first maximum element in a list using a comparable transformation
-}
maximumBy : (a -> comparable) -> List a -> Maybe a
maximumBy f ls =
let maxBy x (y, fy) = let fx = f x in if fx > fy then (x, fx) else (y, fy)
in case ls of
[l'] -> Just l'
l'::ls' -> Just <| fst <| foldl maxBy (l', f l') ls'
_ -> Nothing
{-| Find the first minimum element in a list using a comparable transformation
-}
minimumBy : (a -> comparable) -> List a -> Maybe a
minimumBy f ls =
let minBy x (y, fy) = let fx = f x in if fx < fy then (x, fx) else (y, fy)
in case ls of
[l'] -> Just l'
l'::ls' -> Just <| fst <| foldl minBy (l', f l') ls'
_ -> Nothing
{-| Take elements in order as long as the predicate evaluates to `True`
-}
takeWhile : (a -> Bool) -> List a -> List a
takeWhile predicate list =
case list of
[] -> []
x::xs -> if (predicate x) then x :: takeWhile predicate xs
else []
{-| Drop elements in order as long as the predicate evaluates to `True`
-}
dropWhile : (a -> Bool) -> List a -> List a
dropWhile predicate list =
case list of
[] -> []
x::xs -> if (predicate x) then dropWhile predicate xs
else list
{-| Drop _all_ duplicate elements from the list
-}
dropDuplicates : List comparable -> List comparable
dropDuplicates list =
let
step next (set, acc) =
if Set.member next set
then (set, acc)
else (Set.insert next set, next::acc)
in
List.foldl step (Set.empty, []) list |> snd |> List.reverse
{-| Map functions taking multiple arguments over multiple lists. Each list should be of the same length.
( (\a b c -> a + b * c)
`map` [1,2,3]
`andMap` [4,5,6]
`andMap` [2,1,1]
) == [9,7,9]
-}
andMap : List (a -> b) -> List a -> List b
andMap fl l = map2 (<|) fl l
{-| Equivalent to `concatMap` with arguments reversed. Ideal to use as an infix function, chaining together functions that return List. For example, suppose you want to have a cartesian product of [1,2] and [3,4]:
[1,2] `andThen` \x ->
[3,4] `andThen` \y ->
[(x,y)]
will give back the list:
[(1,3),(1,4),(2,3),(2,4)]
Now suppose we want to have a cartesian product between the first list and the second list and its doubles:
[1,2] `andThen` \x ->
[3,4] `andThen` \y ->
[y,y*2] `andThen` \z ->
[(x,z)]
will give back the list:
[(1,3),(1,6),(1,4),(1,8),(2,3),(2,6),(2,4),(2,8)]
Advanced functional programmers will recognize this as the implementation of bind operator (>>=) for lists from the `Monad` typeclass.
-}
andThen : List a -> (a -> List b) -> List b
andThen = flip concatMap
{-| Negation of `member`.
1 `notMember` [1,2,3] == False
4 `notMember` [1,2,3] == True
-}
notMember : a -> List a -> Bool
notMember x = not << member x
{-| Find the first element that satisfies a predicate and return
Just that element. If none match, return Nothing.
find (\num -> num > 5) [2, 4, 6, 8] == Just 6
-}
find : (a -> Bool) -> List a -> Maybe a
find predicate list =
case list of
[] ->
Nothing
first::rest ->
if predicate first then
Just first
else
find predicate rest
{-| Return the index of the first occurrence of the element. Otherwise, return `Nothing`. Indexing starts from 0.
elemIndex 1 [1,2,3] == Just 0
elemIndex 4 [1,2,3] == Nothing
elemIndex 1 [1,2,1] == Just 0
-}
elemIndex : a -> List a -> Maybe Int
elemIndex x = findIndex ((==)x)
{-| Return all indices of occurrences of the element. If element is not found, return empty list. Indexing starts from 0.
elemIndices 1 [1,2,3] == [0]
elemIndices 4 [1,2,3] == []
elemIndices 1 [1,2,1] == [0,2]
-}
elemIndices : a -> List a -> List Int
elemIndices x = findIndices ((==)x)
{-| Take a predicate and a list, return the index of the first element that satisfies the predicate. Otherwise, return `Nothing`. Indexing starts from 0.
findIndex isEven [1,2,3] == Just 1
findIndex isEven [1,3,5] == Nothing
findIndex isEven [1,2,4] == Just 1
-}
findIndex : (a -> Bool) -> List a -> Maybe Int
findIndex p = head << findIndices p
{-| Take a predicate and a list, return indices of all elements satisfying the predicate. Otherwise, return empty list. Indexing starts from 0.
findIndices isEven [1,2,3] == [1]
findIndices isEven [1,3,5] == []
findIndices isEven [1,2,4] == [1,2]
-}
findIndices : (a -> Bool) -> List a -> List Int
findIndices p = map fst << filter (\(i,x) -> p x) << indexedMap (,)
{-| Replace all values that satisfy a predicate with a replacement value.
-}
replaceIf : (a -> Bool) -> a -> List a -> List a
replaceIf predicate replacement list =
List.map (\item -> if predicate item then replacement else item) list
{-| Convert a value to a list containing one value.
singleton 3 == [3]
-}
singleton : a -> List a
singleton x = [x]
{-| Take a predicate and a list, and return a list that contains elements which fails to satisfy the predicate.
This is equivalent to `List.filter (not << predicate) list`.
removeWhen isEven [1,2,3,4] == [1,3]
-}
removeWhen : (a -> Bool) -> List a -> List a
removeWhen pred list =
List.filter (not << pred) list
{-| Take a list and a list of lists, insert that list between every list in the list of lists, concatenate the result. `intercalate xs xss` is equivalent to `concat (intersperse xs xss)`.
intercalate [0,0] [[1,2],[3,4],[5,6]] == [1,2,0,0,3,4,0,0,5,6]
-}
intercalate : List a -> List (List a) -> List a
intercalate xs = concat << intersperse xs
{-| Transpose rows and columns of the list of lists.
transpose [[1,2,3],[4,5,6]] == [[1,4],[2,5],[3,6]]
If some rows are shorter than the following rows, their elements are skipped:
transpose [[10,11],[20],[],[30,31,32]] == [[10,20,30],[11,31],[32]]
-}
transpose : List (List a) -> List (List a)
transpose ll =
case ll of
[] -> []
([]::xss) -> transpose xss
((x::xs)::xss) ->
let
heads = filterMap head xss
tails = filterMap tail xss
in
(x::heads)::transpose (xs::tails)
{-| Return the list of all subsequences of a list.
subsequences [1,2,3] == [[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
-}
subsequences : List a -> List (List a)
subsequences xs = []::subsequencesNonEmpty xs
{-| Return the list of all subsequences of the argument, except for the empty list.
subsequencesNonEmpty [1,2,3] == [[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
-}
subsequencesNonEmpty : List a -> List (List a)
subsequencesNonEmpty xs =
case xs of
[] -> []
(x::xs) ->
let f ys r = ys::(x::ys)::r
in [x]::foldr f [] (subsequencesNonEmpty xs)
{-| Return the list of of all permutations of a list. The result is in lexicographic order.
permutations [1,2,3] == [[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
-}
permutations : List a -> List (List a)
permutations xs' =
case xs' of
[] -> [[]]
xs -> let f (y,ys) = map ((::)y) (permutations ys)
in concatMap f (select xs)
{-| Return a list that contains elements from the two provided, in alternate order.
If one list runs out of items, append the items from the remaining list.
interweave [1,3] [2,4] == [1,2,3,4]
interweave [1,3,5,7] [2,4] == [1,2,3,4,5,7]
-}
interweave : List a -> List a -> List a
interweave l1 l2 =
interweaveHelp l1 l2 []
interweaveHelp : List a -> List a -> List a -> List a
interweaveHelp l1 l2 acc =
case (l1, l2) of
(x::xs, y::ys) ->
interweaveHelp xs ys <| acc ++ [x, y]
(x, []) ->
acc ++ x
([], y) ->
acc ++ y
{-| Remove all duplicates from a list and return a list of distinct elements.
-}
unique : List comparable -> List comparable
unique list =
uniqueHelp Set.empty list
uniqueHelp : Set comparable -> List comparable -> List comparable
uniqueHelp existing remaining =
case remaining of
[] ->
[]
first :: rest ->
if Set.member first existing then
uniqueHelp existing rest
else
first :: uniqueHelp (Set.insert first existing) rest
{-| Variant of `foldl` that has no starting value argument and treats the head of the list as its starting value. If the list is empty, return `Nothing`.
foldl1 max [1,2,3,2,1] == Just 3
foldl1 max [] == Nothing
foldl1 (-) [1,2,3] == Just -4
-}
foldl1 : (a -> a -> a) -> List a -> Maybe a
foldl1 f xs =
let
mf x m = Just (case m of
Nothing -> x
Just y -> f y x)
in
List.foldl mf Nothing xs
{-| Variant of `foldr` that has no starting value argument and treats the last element of the list as its starting value. If the list is empty, return `Nothing`.
foldr1 min [1,2,3,2,1] == Just 1
foldr1 min [] == Nothing
foldr1 (-) [1,2,3] == Just 2
-}
foldr1 : (a -> a -> a) -> List a -> Maybe a
foldr1 f xs =
let
mf x m = Just (case m of
Nothing -> x
Just y -> f x y)
in
List.foldr mf Nothing xs
{-| `scanl1` is a variant of `scanl` that has no starting value argument.
Compare:
List.scanl (+) 0 [1,2,3] == [0,1,3,6]
scanl1 (+) [1,2,3] == [1,3,6]
List.scanl (-) 0 [1,2,3] == [0,1,1,2]
scanl1 (-) [1,2,3] == [1,1,2]
List.scanl (flip (-)) 0 [1,2,3] == [0,-1,-3,-6]
scanl1 (flip (-)) [1,2,3] == [1,-1,4]
-}
scanl1 : (a -> a -> a) -> List a -> List a
scanl1 f xs' =
case xs' of
[] -> []
(x::xs) -> scanl f x xs
{-| `scanr` is a right-to-left dual of `scanl`. Note that:
head (scanr f z xs) == foldr f z xs
Examples:
scanr (+) 0 [1,2,3] == [6,5,3,0]
scanr (-) 0 [1,2,3] == [2,-1,3,0]
-}
scanr : (a -> b -> b) -> b -> List a -> List b
scanr f acc xs' =
case xs' of
[] -> [acc]
(x::xs) ->
case scanr f acc xs of
(q::_) as qs ->
f x q :: qs
[] ->
[]
{-| `scanr1` is a variant of `scanr` that has no starting value argument.
scanr1 (+) [1,2,3] == [6,5,3]
scanr1 (-) [1,2,3] == [2,-1,3]
scanr1 (flip (-)) [1,2,3] == [0,1,3]
-}
scanr1 : (a -> a -> a) -> List a -> List a
scanr1 f xs' =
case xs' of
[] -> []
[x] -> [x]
(x::xs) ->
case scanr1 f xs of
(q::_) as qs ->
f x q :: qs
[] ->
[]
{-| The `unfoldr` function is "dual" to `foldr`. `foldr` reduces a list to a summary value, `unfoldr` builds a list from a seed. The function takes a function and a starting element. It applies the function to the element. If the result is `Just (a, b)`, `a` is accumulated and the function is applied to `b`. If the result is `Nothing`, the list accumulated so far is returned.
unfoldr (\b -> if b == 0 then Nothing else Just (b, b-1)) 5 == [5,4,3,2,1]
-}
unfoldr : (b -> Maybe (a, b)) -> b -> List a
unfoldr f seed =
case f seed of
Nothing -> []
Just (a, b) -> a :: unfoldr f b
{-| Take a number and a list, return a tuple of lists, where first part is prefix of the list of length equal the number, and second part is the remainder of the list. `splitAt n xs` is equivalent to `(take n xs, drop n xs)`.
splitAt 3 [1,2,3,4,5] == ([1,2,3],[4,5])
splitAt 1 [1,2,3] == ([1],[2,3])
splitAt 3 [1,2,3] == ([1,2,3],[])
splitAt 4 [1,2,3] == ([1,2,3],[])
splitAt 0 [1,2,3] == ([],[1,2,3])
splitAt (-1) [1,2,3] == ([],[1,2,3])
-}
splitAt : Int -> List a -> (List a, List a)
splitAt n xs = (take n xs, drop n xs)
{-| Take elements from the end, while predicate still holds.
takeWhileEnd ((<)5) [1..10] == [6,7,8,9,10]
-}
takeWhileEnd : (a -> Bool) -> List a -> List a
takeWhileEnd p =
let
step x (xs,free) = if p x && free then (x::xs,True) else (xs, False)
in
fst << foldr step ([], True)
{-| Drop elements from the end, while predicate still holds.
dropWhileEnd ((<)5) [1..10] == [1,2,3,4,5]
-}
dropWhileEnd : (a -> Bool) -> List a -> List a
dropWhileEnd p = foldr (\x xs -> if p x && isEmpty xs then [] else x::xs) []
{-| Take a predicate and a list, return a tuple. The first part of the tuple is longest prefix of that list, for each element of which the predicate holds. The second part of the tuple is the remainder of the list. `span p xs` is equivalent to `(takeWhile p xs, dropWhile p xs)`.
span (< 3) [1,2,3,4,1,2,3,4] == ([1,2],[3,4,1,2,3,4])
span (< 5) [1,2,3] == ([1,2,3],[])
span (< 0) [1,2,3] == ([],[1,2,3])
-}
span : (a -> Bool) -> List a -> (List a, List a)
span p xs = (takeWhile p xs, dropWhile p xs)
{-| Take a predicate and a list, return a tuple. The first part of the tuple is longest prefix of that list, for each element of which the predicate *does not* hold. The second part of the tuple is the remainder of the list. `span p xs` is equivalent to `(dropWhile p xs, takeWhile p xs)`.
break (> 3) [1,2,3,4,1,2,3,4] == ([1,2,3],[4,1,2,3,4])
break (< 5) [1,2,3] == ([],[1,2,3])
break (> 5) [1,2,3] == ([1,2,3],[])
-}
break : (a -> Bool) -> List a -> (List a, List a)
break p = span (not << p)
{-| Drop the given prefix from the list. If the list doesn't start with that prefix, return `Nothing`.
stripPrefix [1,2] [1,2,3,4] == Just [3,4]
stripPrefix [1,2,3] [1,2,3,4,5] == Just [4,5]
stripPrefix [1,2,3] [1,2,3] == Just []
stripPrefix [1,2,3] [1,2] == Nothing
stripPrefix [3,2,1] [1,2,3,4,5] == Nothing
-}
stripPrefix : List a -> List a -> Maybe (List a)
stripPrefix prefix xs =
let
step e m =
case m of
Nothing -> Nothing
Just [] -> Nothing
Just (x::xs') -> if e == x
then Just xs'
else Nothing
in
foldl step (Just xs) prefix
{-| Group similar elements together. `group` is equivalent to `groupBy (==)`.
group [1,2,2,3,3,3,2,2,1] == [[1],[2,2],[3,3,3],[2,2],[1]]
-}
group : List a -> List (List a)
group = groupBy (==)
{-| Group elements together, using a custom equality test.
groupBy (\x y -> fst x == fst y) [(0,'a'),(0,'b'),(1,'c'),(1,'d')] == [[(0,'a'),(0,'b')],[(1,'c'),(1,'d')]]
The equality test should be an equivalent relationship, i.e. it should have the properties of reflexivity, symmetry, and transitivity. For non-equivalent relations it gives non-intuitive behavior:
groupBy (<) [1,2,3,2,4,1,3,2,1] == [[1,2,3,2,4],[1,3,2],[1]]
For grouping elements with a comparison test, which must only hold the property of transitivity, see `groupByTransitive`.
-}
groupBy : (a -> a -> Bool) -> List a -> List (List a)
groupBy eq xs' =
case xs' of
[] -> []
(x::xs) -> let (ys,zs) = span (eq x) xs
in (x::ys)::groupBy eq zs
{-| Group elements together, using a custom comparison test. Start a new group each time the comparison test doesn't hold for two adjacent elements.
groupByTransitive (<) [1,2,3,2,4,1,3,2,1] == [[1,2,3],[2,4],[1,3],[2],[1]]
-}
groupByTransitive : (a -> a -> Bool) -> List a -> List (List a)
groupByTransitive cmp xs' =
case xs' of
[] -> []
[x] -> [[x]]
(x::((x'::_) as xs)) ->
case groupByTransitive cmp xs of
(y::ys) as r ->
if cmp x x'
then (x::y)::ys
else [x]::r
[] ->
[]
{-| Return all initial segments of a list, from shortest to longest, empty list first, the list itself last.
inits [1,2,3] == [[],[1],[1,2],[1,2,3]]
-}
inits : List a -> List (List a)
inits = foldr (\e acc -> []::map ((::)e) acc) [[]]
{-| Return all final segments of a list, from longest to shortest, the list itself first, empty list last.
tails [1,2,3] == [[1,2,3],[2,3],[3],[]]
-}
tails : List a -> List (List a)
tails = foldr tailsHelp [[]]
tailsHelp : a -> List (List a) -> List (List a)
tailsHelp e list =
case list of
(x::xs) ->
(e::x)::x::xs
[] ->
[]
{-| Return all combinations in the form of (element, rest of the list). Read [Haskell Libraries proposal](https://mail.haskell.org/pipermail/libraries/2008-February/009270.html) for further ideas on how to use this function.
select [1,2,3,4] == [(1,[2,3,4]),(2,[1,3,4]),(3,[1,2,4]),(4,[1,2,3])]
-}
select : List a -> List (a, List a)
select xs =
case xs of
[] -> []
(x::xs) -> (x,xs)::map (\(y,ys) -> (y,x::ys)) (select xs)
{-| Return all combinations in the form of (elements before, element, elements after).
selectSplit [1,2,3] == [([],1,[2,3]),([1],2,[3]),([1,2],3,[])]
-}
selectSplit : List a -> List (List a, a, List a)
selectSplit xs =
case xs of
[] -> []
(x::xs) -> ([],x,xs)::map (\(lys,y,rys) -> (x::lys,y,rys)) (selectSplit xs)
{-| Take 2 lists and return True, if the first list is the prefix of the second list.
-}
isPrefixOf : List a -> List a -> Bool
isPrefixOf prefix = all identity << map2 (==) prefix
{-| Take 2 lists and return True, if the first list is the suffix of the second list.
-}
isSuffixOf : List a -> List a -> Bool
isSuffixOf suffix xs = isPrefixOf (reverse suffix) (reverse xs)
{-| Take 2 lists and return True, if the first list is an infix of the second list.
-}
isInfixOf : List a -> List a -> Bool
isInfixOf infix xs = any (isPrefixOf infix) (tails xs)
{-| Take 2 lists and return True, if the first list is a subsequence of the second list.
-}
isSubsequenceOf : List a -> List a -> Bool
isSubsequenceOf subseq xs = subseq `member` subsequences xs
{-| Take 2 lists and return True, if the first list is a permutation of the second list.
-}
isPermutationOf : List a -> List a -> Bool
isPermutationOf permut xs = permut `member` permutations xs
{-| Take two lists and returns a list of corresponding pairs
-}
zip : List a -> List b -> List (a,b)
zip = map2 (,)
{-| Take three lists and returns a list of triples
-}
zip3 : List a -> List b -> List c -> List (a,b,c)
zip3 = map3 (,,)
{-| Take four lists and returns a list of quadruples
-}
zip4 : List a -> List b -> List c -> List d -> List (a,b,c,d)
zip4 = map4 (,,,)
{-| Take five lists and returns a list of quintuples
-}
zip5 : List a -> List b -> List c -> List d -> List e -> List (a,b,c,d,e)
zip5 = map5 (,,,,)
{-| Map functions taking multiple arguments over multiple lists, regardless of list length.
All possible combinations will be explored.
lift2 (+) [1,2,3] [4,5] == [5,6,6,7,7,8]
-}
lift2 : (a -> b -> c) -> List a -> List b -> List c
lift2 f la lb =
la `andThen` (\a -> lb `andThen` (\b -> [f a b]))
{-|
-}
lift3 : (a -> b -> c -> d) -> List a -> List b -> List c -> List d
lift3 f la lb lc =
la `andThen` (\a -> lb `andThen` (\b -> lc `andThen` (\c -> [f a b c])))
{-|
-}
lift4 : (a -> b -> c -> d -> e) -> List a -> List b -> List c -> List d -> List e
lift4 f la lb lc ld =
la `andThen` (\a -> lb `andThen` (\b -> lc `andThen` (\c -> ld `andThen` (\d -> [f a b c d]))))
```