2020-05-18 15:59:07 +03:00
|
|
|
module Data.List
|
|
|
|
|
2020-07-29 12:51:07 +03:00
|
|
|
import Data.Nat
|
2020-08-04 22:03:18 +03:00
|
|
|
import Data.List1
|
2020-12-04 04:49:26 +03:00
|
|
|
import Data.Fin
|
2020-07-29 12:51:07 +03:00
|
|
|
|
2020-12-18 13:33:56 +03:00
|
|
|
%default total
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
public export
|
|
|
|
isNil : List a -> Bool
|
2020-07-08 02:55:55 +03:00
|
|
|
isNil [] = True
|
|
|
|
isNil _ = False
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
public export
|
|
|
|
isCons : List a -> Bool
|
2020-07-08 02:55:55 +03:00
|
|
|
isCons [] = False
|
|
|
|
isCons _ = True
|
2020-05-18 15:59:07 +03:00
|
|
|
|
2020-06-01 10:28:37 +03:00
|
|
|
public export
|
|
|
|
snoc : List a -> a -> List a
|
|
|
|
snoc xs x = xs ++ [x]
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
public export
|
|
|
|
take : Nat -> List a -> List a
|
|
|
|
take (S k) (x :: xs) = x :: take k xs
|
2020-06-30 02:32:18 +03:00
|
|
|
take _ _ = []
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
public export
|
|
|
|
drop : (n : Nat) -> (xs : List a) -> List a
|
|
|
|
drop Z xs = xs
|
|
|
|
drop (S n) [] = []
|
2020-07-08 02:55:55 +03:00
|
|
|
drop (S n) (_::xs) = drop n xs
|
2020-05-18 15:59:07 +03:00
|
|
|
|
2020-06-12 11:50:06 +03:00
|
|
|
||| Satisfiable if `k` is a valid index into `xs`
|
|
|
|
|||
|
|
|
|
||| @ k the potential index
|
|
|
|
||| @ xs the list into which k may be an index
|
|
|
|
public export
|
|
|
|
data InBounds : (k : Nat) -> (xs : List a) -> Type where
|
|
|
|
||| Z is a valid index into any cons cell
|
|
|
|
InFirst : InBounds Z (x :: xs)
|
|
|
|
||| Valid indices can be extended
|
|
|
|
InLater : InBounds k xs -> InBounds (S k) (x :: xs)
|
|
|
|
|
2020-07-01 21:37:40 +03:00
|
|
|
public export
|
2020-06-12 11:50:06 +03:00
|
|
|
Uninhabited (InBounds k []) where
|
|
|
|
uninhabited InFirst impossible
|
|
|
|
uninhabited (InLater _) impossible
|
|
|
|
|
|
|
|
||| Decide whether `k` is a valid index into `xs`
|
2020-07-01 21:37:40 +03:00
|
|
|
public export
|
2020-06-12 11:50:06 +03:00
|
|
|
inBounds : (k : Nat) -> (xs : List a) -> Dec (InBounds k xs)
|
2020-07-08 02:55:55 +03:00
|
|
|
inBounds _ [] = No uninhabited
|
|
|
|
inBounds Z (_ :: _) = Yes InFirst
|
2020-06-12 11:50:06 +03:00
|
|
|
inBounds (S k) (x :: xs) with (inBounds k xs)
|
|
|
|
inBounds (S k) (x :: xs) | (Yes prf) = Yes (InLater prf)
|
|
|
|
inBounds (S k) (x :: xs) | (No contra)
|
2020-06-30 02:32:18 +03:00
|
|
|
= No $ \(InLater y) => contra y
|
2020-06-12 11:50:06 +03:00
|
|
|
|
|
|
|
||| Find a particular element of a list.
|
|
|
|
|||
|
|
|
|
||| @ ok a proof that the index is within bounds
|
2020-07-01 21:37:40 +03:00
|
|
|
public export
|
2020-12-03 17:26:05 +03:00
|
|
|
index : (n : Nat) -> (xs : List a) -> {auto 0 ok : InBounds n xs} -> a
|
2020-06-12 11:50:06 +03:00
|
|
|
index Z (x :: xs) {ok = InFirst} = x
|
2020-07-08 02:55:55 +03:00
|
|
|
index (S k) (_ :: xs) {ok = InLater _} = index k xs
|
2020-06-12 11:50:06 +03:00
|
|
|
|
2020-12-04 04:49:26 +03:00
|
|
|
public export
|
|
|
|
index' : (xs : List a) -> Fin (length xs) -> a
|
|
|
|
index' (x::_) FZ = x
|
|
|
|
index' (_::xs) (FS i) = index' xs i
|
|
|
|
|
2020-05-20 17:36:50 +03:00
|
|
|
||| Generate a list by repeatedly applying a partial function until exhausted.
|
|
|
|
||| @ f the function to iterate
|
|
|
|
||| @ x the initial value that will be the head of the list
|
2020-12-18 13:33:56 +03:00
|
|
|
covering
|
2020-05-20 17:36:50 +03:00
|
|
|
public export
|
|
|
|
iterate : (f : a -> Maybe a) -> (x : a) -> List a
|
|
|
|
iterate f x = x :: case f x of
|
|
|
|
Nothing => []
|
|
|
|
Just y => iterate f y
|
|
|
|
|
2020-12-18 23:54:23 +03:00
|
|
|
covering
|
|
|
|
public export
|
|
|
|
unfoldr : (b -> Maybe (a, b)) -> b -> List a
|
|
|
|
unfoldr f c = case f c of
|
|
|
|
Nothing => []
|
|
|
|
Just (a, n) => a :: unfoldr f n
|
|
|
|
|
2020-12-18 13:33:56 +03:00
|
|
|
public export
|
|
|
|
iterateN : Nat -> (a -> a) -> a -> List a
|
|
|
|
iterateN Z _ _ = []
|
|
|
|
iterateN (S n) f x = x :: iterateN n f (f x)
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
public export
|
|
|
|
takeWhile : (p : a -> Bool) -> List a -> List a
|
|
|
|
takeWhile p [] = []
|
|
|
|
takeWhile p (x::xs) = if p x then x :: takeWhile p xs else []
|
|
|
|
|
|
|
|
public export
|
|
|
|
dropWhile : (p : a -> Bool) -> List a -> List a
|
|
|
|
dropWhile p [] = []
|
|
|
|
dropWhile p (x::xs) = if p x then dropWhile p xs else x::xs
|
|
|
|
|
|
|
|
public export
|
|
|
|
filter : (p : a -> Bool) -> List a -> List a
|
|
|
|
filter p [] = []
|
|
|
|
filter p (x :: xs)
|
|
|
|
= if p x
|
|
|
|
then x :: filter p xs
|
|
|
|
else filter p xs
|
|
|
|
|
2020-05-27 13:21:29 +03:00
|
|
|
||| Find the first element of the list that satisfies the predicate.
|
|
|
|
public export
|
|
|
|
find : (p : a -> Bool) -> (xs : List a) -> Maybe a
|
|
|
|
find p [] = Nothing
|
|
|
|
find p (x::xs) = if p x then Just x else find p xs
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
||| Find associated information in a list using a custom comparison.
|
|
|
|
public export
|
|
|
|
lookupBy : (a -> a -> Bool) -> a -> List (a, b) -> Maybe b
|
|
|
|
lookupBy p e [] = Nothing
|
2020-06-30 02:32:18 +03:00
|
|
|
lookupBy p e ((l, r) :: xs) =
|
|
|
|
if p e l then
|
|
|
|
Just r
|
|
|
|
else
|
|
|
|
lookupBy p e xs
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
||| Find associated information in a list using Boolean equality.
|
|
|
|
public export
|
|
|
|
lookup : Eq a => a -> List (a, b) -> Maybe b
|
|
|
|
lookup = lookupBy (==)
|
|
|
|
|
|
|
|
||| Check if something is a member of a list using a custom comparison.
|
|
|
|
public export
|
|
|
|
elemBy : (a -> a -> Bool) -> a -> List a -> Bool
|
|
|
|
elemBy p e [] = False
|
2020-07-08 02:55:55 +03:00
|
|
|
elemBy p e (x::xs) = p e x || elemBy p e xs
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
public export
|
|
|
|
nubBy : (a -> a -> Bool) -> List a -> List a
|
|
|
|
nubBy = nubBy' []
|
|
|
|
where
|
|
|
|
nubBy' : List a -> (a -> a -> Bool) -> List a -> List a
|
|
|
|
nubBy' acc p [] = []
|
|
|
|
nubBy' acc p (x::xs) =
|
|
|
|
if elemBy p x acc then
|
|
|
|
nubBy' acc p xs
|
|
|
|
else
|
|
|
|
x :: nubBy' (x::acc) p xs
|
|
|
|
|
|
|
|
||| O(n^2). The nub function removes duplicate elements from a list. In
|
|
|
|
||| particular, it keeps only the first occurrence of each element. It is a
|
|
|
|
||| special case of nubBy, which allows the programmer to supply their own
|
|
|
|
||| equality test.
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| nub (the (List _) [1,2,1,3])
|
|
|
|
||| ```
|
|
|
|
public export
|
|
|
|
nub : Eq a => List a -> List a
|
|
|
|
nub = nubBy (==)
|
|
|
|
|
|
|
|
||| The deleteBy function behaves like delete, but takes a user-supplied equality predicate.
|
|
|
|
public export
|
|
|
|
deleteBy : (a -> a -> Bool) -> a -> List a -> List a
|
|
|
|
deleteBy _ _ [] = []
|
|
|
|
deleteBy eq x (y::ys) = if x `eq` y then ys else y :: deleteBy eq x ys
|
|
|
|
|
|
|
|
||| `delete x` removes the first occurrence of `x` from its list argument. For
|
|
|
|
||| example,
|
|
|
|
|||
|
|
|
|
|||````idris example
|
|
|
|
|||delete 'a' ['b', 'a', 'n', 'a', 'n', 'a']
|
|
|
|
|||````
|
|
|
|
|||
|
|
|
|
||| It is a special case of deleteBy, which allows the programmer to supply
|
|
|
|
||| their own equality test.
|
|
|
|
public export
|
|
|
|
delete : Eq a => a -> List a -> List a
|
|
|
|
delete = deleteBy (==)
|
|
|
|
|
|
|
|
||| The unionBy function returns the union of two lists by user-supplied equality predicate.
|
|
|
|
public export
|
|
|
|
unionBy : (a -> a -> Bool) -> List a -> List a -> List a
|
|
|
|
unionBy eq xs ys = xs ++ foldl (flip (deleteBy eq)) (nubBy eq ys) xs
|
|
|
|
|
|
|
|
||| Compute the union of two lists according to their `Eq` implementation.
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| union ['d', 'o', 'g'] ['c', 'o', 'w']
|
|
|
|
||| ```
|
|
|
|
|||
|
|
|
|
public export
|
|
|
|
union : Eq a => List a -> List a -> List a
|
|
|
|
union = unionBy (==)
|
|
|
|
|
2020-09-28 15:15:22 +03:00
|
|
|
public export
|
|
|
|
spanBy : (a -> Maybe b) -> List a -> (List b, List a)
|
|
|
|
spanBy p [] = ([], [])
|
|
|
|
spanBy p (x :: xs) = case p x of
|
|
|
|
Nothing => ([], x :: xs)
|
|
|
|
Just y => let (ys, zs) = spanBy p xs in (y :: ys, zs)
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
public export
|
|
|
|
span : (a -> Bool) -> List a -> (List a, List a)
|
|
|
|
span p [] = ([], [])
|
|
|
|
span p (x::xs) =
|
|
|
|
if p x then
|
|
|
|
let (ys, zs) = span p xs in
|
2020-09-28 15:15:22 +03:00
|
|
|
(x::ys, zs)
|
2020-05-18 15:59:07 +03:00
|
|
|
else
|
|
|
|
([], x::xs)
|
|
|
|
|
|
|
|
public export
|
|
|
|
break : (a -> Bool) -> List a -> (List a, List a)
|
|
|
|
break p xs = span (not . p) xs
|
|
|
|
|
|
|
|
public export
|
2020-08-04 22:03:18 +03:00
|
|
|
split : (a -> Bool) -> List a -> List1 (List a)
|
2020-05-18 15:59:07 +03:00
|
|
|
split p xs =
|
|
|
|
case break p xs of
|
2020-09-22 17:07:40 +03:00
|
|
|
(chunk, []) => singleton chunk
|
|
|
|
(chunk, (c :: rest)) => cons chunk (split p (assert_smaller xs rest))
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
public export
|
|
|
|
splitAt : (n : Nat) -> (xs : List a) -> (List a, List a)
|
|
|
|
splitAt Z xs = ([], xs)
|
|
|
|
splitAt (S k) [] = ([], [])
|
|
|
|
splitAt (S k) (x :: xs)
|
|
|
|
= let (tk, dr) = splitAt k xs in
|
|
|
|
(x :: tk, dr)
|
|
|
|
|
|
|
|
public export
|
|
|
|
partition : (a -> Bool) -> List a -> (List a, List a)
|
|
|
|
partition p [] = ([], [])
|
|
|
|
partition p (x::xs) =
|
|
|
|
let (lefts, rights) = partition p xs in
|
|
|
|
if p x then
|
|
|
|
(x::lefts, rights)
|
|
|
|
else
|
|
|
|
(lefts, x::rights)
|
|
|
|
|
|
|
|
||| The inits function returns all initial segments of the argument, shortest
|
|
|
|
||| first. For example,
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| inits [1,2,3]
|
|
|
|
||| ```
|
|
|
|
public export
|
|
|
|
inits : List a -> List (List a)
|
|
|
|
inits xs = [] :: case xs of
|
|
|
|
[] => []
|
|
|
|
x :: xs' => map (x ::) (inits xs')
|
|
|
|
|
|
|
|
||| The tails function returns all final segments of the argument, longest
|
|
|
|
||| first. For example,
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| tails [1,2,3] == [[1,2,3], [2,3], [3], []]
|
|
|
|
|||```
|
|
|
|
public export
|
|
|
|
tails : List a -> List (List a)
|
|
|
|
tails xs = xs :: case xs of
|
|
|
|
[] => []
|
|
|
|
_ :: xs' => tails xs'
|
|
|
|
|
|
|
|
||| Split on the given element.
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| splitOn 0 [1,0,2,0,0,3]
|
|
|
|
||| ```
|
|
|
|
|||
|
|
|
|
public export
|
2020-08-04 22:03:18 +03:00
|
|
|
splitOn : Eq a => a -> List a -> List1 (List a)
|
2020-05-18 15:59:07 +03:00
|
|
|
splitOn a = split (== a)
|
|
|
|
|
2020-11-27 22:10:08 +03:00
|
|
|
public export
|
|
|
|
replaceWhen : (a -> Bool) -> a -> List a -> List a
|
|
|
|
replaceWhen p b l = map (\c => if p c then b else c) l
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
||| Replaces all occurences of the first argument with the second argument in a list.
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| replaceOn '-' ',' ['1', '-', '2', '-', '3']
|
|
|
|
||| ```
|
|
|
|
|||
|
|
|
|
public export
|
|
|
|
replaceOn : Eq a => a -> a -> List a -> List a
|
2020-11-27 22:10:08 +03:00
|
|
|
replaceOn a = replaceWhen (== a)
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
public export
|
|
|
|
reverseOnto : List a -> List a -> List a
|
|
|
|
reverseOnto acc [] = acc
|
|
|
|
reverseOnto acc (x::xs) = reverseOnto (x::acc) xs
|
|
|
|
|
|
|
|
public export
|
|
|
|
reverse : List a -> List a
|
|
|
|
reverse = reverseOnto []
|
|
|
|
|
|
|
|
||| Construct a list with `n` copies of `x`.
|
|
|
|
||| @ n how many copies
|
|
|
|
||| @ x the element to replicate
|
|
|
|
public export
|
|
|
|
replicate : (n : Nat) -> (x : a) -> List a
|
|
|
|
replicate Z _ = []
|
|
|
|
replicate (S n) x = x :: replicate n x
|
|
|
|
|
|
|
|
||| Compute the intersect of two lists by user-supplied equality predicate.
|
|
|
|
export
|
|
|
|
intersectBy : (a -> a -> Bool) -> List a -> List a -> List a
|
|
|
|
intersectBy eq xs ys = [x | x <- xs, any (eq x) ys]
|
|
|
|
|
|
|
|
||| Compute the intersect of two lists according to the `Eq` implementation for the elements.
|
|
|
|
export
|
|
|
|
intersect : Eq a => List a -> List a -> List a
|
|
|
|
intersect = intersectBy (==)
|
|
|
|
|
2020-09-14 18:07:25 +03:00
|
|
|
export
|
|
|
|
intersectAllBy : (a -> a -> Bool) -> List (List a) -> List a
|
|
|
|
intersectAllBy eq [] = []
|
|
|
|
intersectAllBy eq (xs :: xss) = filter (\x => all (elemBy eq x) xss) xs
|
|
|
|
|
|
|
|
export
|
|
|
|
intersectAll : Eq a => List (List a) -> List a
|
|
|
|
intersectAll = intersectAllBy (==)
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
||| Combine two lists elementwise using some function.
|
|
|
|
|||
|
|
|
|
||| If the lists are different lengths, the result is truncated to the
|
|
|
|
||| length of the shortest list.
|
|
|
|
export
|
|
|
|
zipWith : (a -> b -> c) -> List a -> List b -> List c
|
|
|
|
zipWith _ [] _ = []
|
|
|
|
zipWith _ _ [] = []
|
|
|
|
zipWith f (x::xs) (y::ys) = f x y :: zipWith f xs ys
|
|
|
|
|
|
|
|
||| Combine two lists elementwise into pairs.
|
|
|
|
|||
|
|
|
|
||| If the lists are different lengths, the result is truncated to the
|
|
|
|
||| length of the shortest list.
|
|
|
|
export
|
|
|
|
zip : List a -> List b -> List (a, b)
|
|
|
|
zip = zipWith \x, y => (x, y)
|
|
|
|
|
|
|
|
export
|
|
|
|
zipWith3 : (a -> b -> c -> d) -> List a -> List b -> List c -> List d
|
|
|
|
zipWith3 _ [] _ _ = []
|
|
|
|
zipWith3 _ _ [] _ = []
|
|
|
|
zipWith3 _ _ _ [] = []
|
|
|
|
zipWith3 f (x::xs) (y::ys) (z::zs) = f x y z :: zipWith3 f xs ys zs
|
|
|
|
|
|
|
|
||| Combine three lists elementwise into tuples.
|
|
|
|
|||
|
|
|
|
||| If the lists are different lengths, the result is truncated to the
|
|
|
|
||| length of the shortest list.
|
|
|
|
export
|
|
|
|
zip3 : List a -> List b -> List c -> List (a, b, c)
|
|
|
|
zip3 = zipWith3 \x, y, z => (x, y, z)
|
|
|
|
|
|
|
|
public export
|
|
|
|
data NonEmpty : (xs : List a) -> Type where
|
|
|
|
IsNonEmpty : NonEmpty (x :: xs)
|
|
|
|
|
|
|
|
export
|
|
|
|
Uninhabited (NonEmpty []) where
|
|
|
|
uninhabited IsNonEmpty impossible
|
|
|
|
|
|
|
|
||| Get the head of a non-empty list.
|
|
|
|
||| @ ok proof the list is non-empty
|
|
|
|
public export
|
2020-07-30 08:01:42 +03:00
|
|
|
head : (l : List a) -> {auto 0 ok : NonEmpty l} -> a
|
2020-05-18 15:59:07 +03:00
|
|
|
head [] impossible
|
2020-07-08 02:55:55 +03:00
|
|
|
head (x :: _) = x
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
||| Get the tail of a non-empty list.
|
|
|
|
||| @ ok proof the list is non-empty
|
|
|
|
public export
|
2020-07-30 08:01:42 +03:00
|
|
|
tail : (l : List a) -> {auto 0 ok : NonEmpty l} -> List a
|
2020-05-18 15:59:07 +03:00
|
|
|
tail [] impossible
|
2020-06-30 02:32:18 +03:00
|
|
|
tail (_ :: xs) = xs
|
2020-05-18 15:59:07 +03:00
|
|
|
|
2020-05-20 17:36:50 +03:00
|
|
|
||| Retrieve the last element of a non-empty list.
|
|
|
|
||| @ ok proof that the list is non-empty
|
|
|
|
public export
|
2020-07-30 08:01:42 +03:00
|
|
|
last : (l : List a) -> {auto 0 ok : NonEmpty l} -> a
|
2020-05-20 17:36:50 +03:00
|
|
|
last [] impossible
|
|
|
|
last [x] = x
|
2020-09-22 17:07:40 +03:00
|
|
|
last (_::x::xs) = List.last (x::xs)
|
2020-05-20 17:36:50 +03:00
|
|
|
|
|
|
|
||| Return all but the last element of a non-empty list.
|
|
|
|
||| @ ok proof the list is non-empty
|
|
|
|
public export
|
2020-07-30 08:01:42 +03:00
|
|
|
init : (l : List a) -> {auto 0 ok : NonEmpty l} -> List a
|
2020-05-20 17:36:50 +03:00
|
|
|
init [] impossible
|
2020-06-30 02:32:18 +03:00
|
|
|
init [_] = []
|
2020-05-20 17:36:50 +03:00
|
|
|
init (x::y::ys) = x :: init (y::ys)
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
||| Attempt to get the head of a list. If the list is empty, return `Nothing`.
|
2020-06-10 21:43:38 +03:00
|
|
|
public export
|
2020-05-18 15:59:07 +03:00
|
|
|
head' : List a -> Maybe a
|
|
|
|
head' [] = Nothing
|
2020-06-30 02:32:18 +03:00
|
|
|
head' (x::_) = Just x
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
||| Attempt to get the tail of a list. If the list is empty, return `Nothing`.
|
|
|
|
export
|
|
|
|
tail' : List a -> Maybe (List a)
|
|
|
|
tail' [] = Nothing
|
2020-06-30 02:32:18 +03:00
|
|
|
tail' (_::xs) = Just xs
|
2020-05-18 15:59:07 +03:00
|
|
|
|
2020-05-20 17:36:50 +03:00
|
|
|
||| Attempt to retrieve the last element of a non-empty list.
|
|
|
|
|||
|
|
|
|
||| If the list is empty, return `Nothing`.
|
|
|
|
export
|
|
|
|
last' : List a -> Maybe a
|
|
|
|
last' [] = Nothing
|
|
|
|
last' xs@(_::_) = Just (last xs)
|
|
|
|
|
|
|
|
||| Attempt to return all but the last element of a non-empty list.
|
|
|
|
|||
|
|
|
|
||| If the list is empty, return `Nothing`.
|
|
|
|
export
|
|
|
|
init' : List a -> Maybe (List a)
|
|
|
|
init' [] = Nothing
|
|
|
|
init' xs@(_::_) = Just (init xs)
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
||| Convert any Foldable structure to a list.
|
|
|
|
export
|
|
|
|
toList : Foldable t => t a -> List a
|
|
|
|
toList = foldr (::) []
|
|
|
|
|
|
|
|
||| Prefix every element in the list with the given element
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| with List (mergeReplicate '>' ['a', 'b', 'c', 'd', 'e'])
|
|
|
|
||| ```
|
|
|
|
|||
|
|
|
|
export
|
|
|
|
mergeReplicate : a -> List a -> List a
|
|
|
|
mergeReplicate sep [] = []
|
|
|
|
mergeReplicate sep (y::ys) = sep :: y :: mergeReplicate sep ys
|
|
|
|
|
|
|
|
|
|
|
|
||| Insert some separator between the elements of a list.
|
|
|
|
|||
|
|
|
|
||| ````idris example
|
|
|
|
||| with List (intersperse ',' ['a', 'b', 'c', 'd', 'e'])
|
|
|
|
||| ````
|
|
|
|
|||
|
|
|
|
export
|
|
|
|
intersperse : a -> List a -> List a
|
|
|
|
intersperse sep [] = []
|
|
|
|
intersperse sep (x::xs) = x :: mergeReplicate sep xs
|
|
|
|
|
|
|
|
||| Given a separator list and some more lists, produce a new list by
|
|
|
|
||| placing the separator between each of the lists.
|
|
|
|
|||
|
|
|
|
||| @ sep the separator
|
|
|
|
||| @ xss the lists between which the separator will be placed
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| intercalate [0, 0, 0] [ [1, 2, 3], [4, 5, 6], [7, 8, 9] ]
|
|
|
|
||| ```
|
|
|
|
export
|
|
|
|
intercalate : (sep : List a) -> (xss : List (List a)) -> List a
|
|
|
|
intercalate sep xss = concat $ intersperse sep xss
|
|
|
|
|
|
|
|
||| Apply a partial function to the elements of a list, keeping the ones at which
|
|
|
|
||| it is defined.
|
2020-06-10 21:43:38 +03:00
|
|
|
public export
|
2020-05-18 15:59:07 +03:00
|
|
|
mapMaybe : (a -> Maybe b) -> List a -> List b
|
|
|
|
mapMaybe f [] = []
|
|
|
|
mapMaybe f (x::xs) =
|
|
|
|
case f x of
|
|
|
|
Nothing => mapMaybe f xs
|
|
|
|
Just j => j :: mapMaybe f xs
|
|
|
|
|
2020-10-02 14:22:57 +03:00
|
|
|
||| Extract all of the values contained in a List of Maybes
|
|
|
|
public export
|
|
|
|
catMaybes : List (Maybe a) -> List a
|
|
|
|
catMaybes = mapMaybe id
|
|
|
|
|
2020-05-20 21:09:14 +03:00
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
-- Special folds
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
||| Foldl a non-empty list without seeding the accumulator.
|
|
|
|
||| @ ok proof that the list is non-empty
|
|
|
|
public export
|
2020-07-30 08:01:42 +03:00
|
|
|
foldl1 : (a -> a -> a) -> (l : List a) -> {auto 0 ok : NonEmpty l} -> a
|
2020-05-20 21:09:14 +03:00
|
|
|
foldl1 f [] impossible
|
|
|
|
foldl1 f (x::xs) = foldl f x xs
|
|
|
|
|
|
|
|
||| Foldr a non-empty list without seeding the accumulator.
|
|
|
|
||| @ ok proof that the list is non-empty
|
|
|
|
public export
|
2020-07-30 08:01:42 +03:00
|
|
|
foldr1 : (a -> a -> a) -> (l : List a) -> {auto 0 ok : NonEmpty l} -> a
|
2020-05-20 21:09:14 +03:00
|
|
|
foldr1 f [] impossible
|
|
|
|
foldr1 f [x] = x
|
2020-09-22 17:07:40 +03:00
|
|
|
foldr1 f (x::y::ys) = f x (List.foldr1 f (y::ys))
|
2020-05-20 21:09:14 +03:00
|
|
|
|
|
|
|
||| Foldl without seeding the accumulator. If the list is empty, return `Nothing`.
|
|
|
|
public export
|
|
|
|
foldl1' : (a -> a -> a) -> List a -> Maybe a
|
|
|
|
foldl1' f [] = Nothing
|
2020-09-22 17:07:40 +03:00
|
|
|
foldl1' f xs@(_::_) = Just (List.foldl1 f xs)
|
2020-05-20 21:09:14 +03:00
|
|
|
|
|
|
|
||| Foldr without seeding the accumulator. If the list is empty, return `Nothing`.
|
|
|
|
public export
|
|
|
|
foldr1' : (a -> a -> a) -> List a -> Maybe a
|
|
|
|
foldr1' f [] = Nothing
|
2020-09-22 17:07:40 +03:00
|
|
|
foldr1' f xs@(_::_) = Just (List.foldr1 f xs)
|
2020-05-20 21:09:14 +03:00
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
-- Sorting
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
||| Check whether a list is sorted with respect to the default ordering for the type of its elements.
|
|
|
|
export
|
|
|
|
sorted : Ord a => List a -> Bool
|
2020-06-30 02:32:18 +03:00
|
|
|
sorted (x :: xs @ (y :: _)) = x <= y && sorted xs
|
|
|
|
sorted _ = True
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
||| Merge two sorted lists using an arbitrary comparison
|
|
|
|
||| predicate. Note that the lists must have been sorted using this
|
|
|
|
||| predicate already.
|
|
|
|
export
|
|
|
|
mergeBy : (a -> a -> Ordering) -> List a -> List a -> List a
|
|
|
|
mergeBy order [] right = right
|
|
|
|
mergeBy order left [] = left
|
|
|
|
mergeBy order (x::xs) (y::ys) =
|
|
|
|
-- The code below will emit `y` before `x` whenever `x == y`.
|
|
|
|
-- If you change this, `sortBy` will stop being stable, unless you fix `sortBy`, too.
|
|
|
|
case order x y of
|
|
|
|
LT => x :: mergeBy order xs (y::ys)
|
|
|
|
_ => y :: mergeBy order (x::xs) ys
|
|
|
|
|
|
|
|
||| Merge two sorted lists using the default ordering for the type of their elements.
|
|
|
|
export
|
|
|
|
merge : Ord a => List a -> List a -> List a
|
|
|
|
merge left right = mergeBy compare left right
|
|
|
|
|
|
|
|
||| Sort a list using some arbitrary comparison predicate.
|
|
|
|
|||
|
|
|
|
||| @ cmp how to compare elements
|
|
|
|
||| @ xs the list to sort
|
|
|
|
export
|
|
|
|
sortBy : (cmp : a -> a -> Ordering) -> (xs : List a) -> List a
|
|
|
|
sortBy cmp [] = []
|
|
|
|
sortBy cmp [x] = [x]
|
|
|
|
sortBy cmp xs = let (x, y) = split xs in
|
|
|
|
mergeBy cmp
|
|
|
|
(sortBy cmp (assert_smaller xs x))
|
|
|
|
(sortBy cmp (assert_smaller xs y)) -- not structurally smaller, hence assert
|
|
|
|
where
|
|
|
|
splitRec : List b -> List a -> (List a -> List a) -> (List a, List a)
|
|
|
|
splitRec (_::_::xs) (y::ys) zs = splitRec xs ys (zs . ((::) y))
|
|
|
|
splitRec _ ys zs = (ys, zs [])
|
|
|
|
-- In the above base-case clause, we put `ys` on the LHS to get a stable sort.
|
|
|
|
-- This is because `mergeBy` prefers taking elements from its RHS operand
|
|
|
|
-- if both heads are equal, and all elements in `zs []` precede all elements of `ys`
|
|
|
|
-- in the original list.
|
|
|
|
|
|
|
|
split : List a -> (List a, List a)
|
|
|
|
split xs = splitRec xs xs id
|
|
|
|
|
|
|
|
||| Sort a list using the default ordering for the type of its elements.
|
|
|
|
export
|
|
|
|
sort : Ord a => List a -> List a
|
|
|
|
sort = sortBy compare
|
|
|
|
|
|
|
|
export
|
|
|
|
isPrefixOfBy : (eq : a -> a -> Bool) -> (left, right : List a) -> Bool
|
2020-06-30 02:32:18 +03:00
|
|
|
isPrefixOfBy p [] _ = True
|
|
|
|
isPrefixOfBy p _ [] = False
|
|
|
|
isPrefixOfBy p (x::xs) (y::ys) = p x y && isPrefixOfBy p xs ys
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
||| The isPrefixOf function takes two lists and returns True iff the first list is a prefix of the second.
|
|
|
|
export
|
|
|
|
isPrefixOf : Eq a => List a -> List a -> Bool
|
|
|
|
isPrefixOf = isPrefixOfBy (==)
|
|
|
|
|
|
|
|
export
|
|
|
|
isSuffixOfBy : (a -> a -> Bool) -> List a -> List a -> Bool
|
|
|
|
isSuffixOfBy p left right = isPrefixOfBy p (reverse left) (reverse right)
|
|
|
|
|
|
|
|
||| The isSuffixOf function takes two lists and returns True iff the first list is a suffix of the second.
|
|
|
|
export
|
|
|
|
isSuffixOf : Eq a => List a -> List a -> Bool
|
|
|
|
isSuffixOf = isSuffixOfBy (==)
|
|
|
|
|
|
|
|
||| The isInfixOf function takes two lists and returns True iff the first list
|
|
|
|
||| is contained, wholly and intact, anywhere within the second.
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| isInfixOf ['b','c'] ['a', 'b', 'c', 'd']
|
|
|
|
||| ```
|
|
|
|
||| ```idris example
|
|
|
|
||| isInfixOf ['b','d'] ['a', 'b', 'c', 'd']
|
|
|
|
||| ```
|
|
|
|
|||
|
|
|
|
export
|
|
|
|
isInfixOf : Eq a => List a -> List a -> Bool
|
|
|
|
isInfixOf n h = any (isPrefixOf n) (tails h)
|
|
|
|
|
|
|
|
||| Transposes rows and columns of a list of lists.
|
|
|
|
|||
|
|
|
|
||| ```idris example
|
|
|
|
||| with List transpose [[1, 2], [3, 4]]
|
|
|
|
||| ```
|
|
|
|
|||
|
|
|
|
||| This also works for non square scenarios, thus
|
|
|
|
||| involution does not always hold:
|
|
|
|
|||
|
|
|
|
||| transpose [[], [1, 2]] = [[1], [2]]
|
|
|
|
||| transpose (transpose [[], [1, 2]]) = [[1, 2]]
|
|
|
|
export
|
|
|
|
transpose : List (List a) -> List (List a)
|
|
|
|
transpose [] = []
|
|
|
|
transpose (heads :: tails) = spreadHeads heads (transpose tails) where
|
|
|
|
spreadHeads : List a -> List (List a) -> List (List a)
|
|
|
|
spreadHeads [] tails = tails
|
|
|
|
spreadHeads (head :: heads) [] = [head] :: spreadHeads heads []
|
|
|
|
spreadHeads (head :: heads) (tail :: tails) = (head :: tail) :: spreadHeads heads tails
|
|
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
-- Properties
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
export
|
|
|
|
Uninhabited ([] = Prelude.(::) x xs) where
|
|
|
|
uninhabited Refl impossible
|
|
|
|
|
|
|
|
export
|
|
|
|
Uninhabited (Prelude.(::) x xs = []) where
|
|
|
|
uninhabited Refl impossible
|
2020-06-01 10:28:37 +03:00
|
|
|
|
2020-06-30 04:18:40 +03:00
|
|
|
||| (::) is injective
|
|
|
|
export
|
2020-08-04 22:03:18 +03:00
|
|
|
consInjective : forall x, xs, y, ys .
|
|
|
|
the (List a) (x :: xs) = the (List b) (y :: ys) -> (x = y, xs = ys)
|
2020-06-30 04:18:40 +03:00
|
|
|
consInjective Refl = (Refl, Refl)
|
|
|
|
|
2020-05-18 15:59:07 +03:00
|
|
|
||| The empty list is a right identity for append.
|
|
|
|
export
|
2020-07-08 02:55:55 +03:00
|
|
|
appendNilRightNeutral : (l : List a) -> l ++ [] = l
|
2020-05-18 15:59:07 +03:00
|
|
|
appendNilRightNeutral [] = Refl
|
2020-07-08 02:55:55 +03:00
|
|
|
appendNilRightNeutral (_::xs) = rewrite appendNilRightNeutral xs in Refl
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
||| Appending lists is associative.
|
|
|
|
export
|
2020-07-08 02:55:55 +03:00
|
|
|
appendAssociative : (l, c, r : List a) -> l ++ (c ++ r) = (l ++ c) ++ r
|
2020-05-18 15:59:07 +03:00
|
|
|
appendAssociative [] c r = Refl
|
2020-07-08 02:55:55 +03:00
|
|
|
appendAssociative (_::xs) c r = rewrite appendAssociative xs c r in Refl
|
2020-05-18 15:59:07 +03:00
|
|
|
|
2020-08-04 22:03:18 +03:00
|
|
|
revOnto : (xs, vs : List a) -> reverseOnto xs vs = reverse vs ++ xs
|
2020-07-08 02:55:55 +03:00
|
|
|
revOnto _ [] = Refl
|
2020-05-18 15:59:07 +03:00
|
|
|
revOnto xs (v :: vs)
|
|
|
|
= rewrite revOnto (v :: xs) vs in
|
|
|
|
rewrite appendAssociative (reverse vs) [v] xs in
|
2020-06-12 11:50:06 +03:00
|
|
|
rewrite revOnto [v] vs in Refl
|
2020-05-18 15:59:07 +03:00
|
|
|
|
|
|
|
export
|
|
|
|
revAppend : (vs, ns : List a) -> reverse ns ++ reverse vs = reverse (vs ++ ns)
|
|
|
|
revAppend [] ns = rewrite appendNilRightNeutral (reverse ns) in Refl
|
|
|
|
revAppend (v :: vs) ns
|
|
|
|
= rewrite revOnto [v] vs in
|
|
|
|
rewrite revOnto [v] (vs ++ ns) in
|
|
|
|
rewrite sym (revAppend vs ns) in
|
|
|
|
rewrite appendAssociative (reverse ns) (reverse vs) [v] in
|
|
|
|
Refl
|
2020-07-29 12:51:07 +03:00
|
|
|
|
|
|
|
export
|
|
|
|
dropFusion : (n, m : Nat) -> (l : List t) -> drop n (drop m l) = drop (n+m) l
|
|
|
|
dropFusion Z m l = Refl
|
|
|
|
dropFusion (S n) Z l = rewrite plusZeroRightNeutral n in Refl
|
|
|
|
dropFusion (S n) (S m) [] = Refl
|
|
|
|
dropFusion (S n) (S m) (x::l) = rewrite plusAssociative n 1 m in
|
|
|
|
rewrite plusCommutative n 1 in
|
2020-08-04 22:03:18 +03:00
|
|
|
dropFusion (S n) m l
|
2020-08-26 18:02:26 +03:00
|
|
|
|
|
|
|
export
|
|
|
|
lengthMap : (xs : List a) -> length (map f xs) = length xs
|
|
|
|
lengthMap [] = Refl
|
|
|
|
lengthMap (x :: xs) = cong S (lengthMap xs)
|