summaryrefslogtreecommitdiff
path: root/lec07/JoinList.hs
blob: 7ec30a909068a124ade841956572bc108674f9d2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
module JoinList where

data JoinList m a = Empty
                  | Single m a
                  | Append m (JoinList m a) (JoinList m a)
  deriving (Eq, Show)

-- Exercise 1

tag :: Monoid m => JoinList m a -> m
tag Empty = mempty
tag Single m _ = m
tag Append m _ _ = m

(+++) :: Monoid m => JoinList m a -> JoinList m a -> JoinList m a
(+++) jl1 jl2 = mappend jl1 jl2

-- Exercise 2

indexJ :: (Sized b, Monoid b) => Int -> JoinList b a -> Maybe a
indexJ _ Empty = Nothing
indexJ n (Single _ a)
       | n == 0 = Just a
       | otherwise = Nothing
indexJ n (Append m jl1 jl2)
       | n < 0 || n >= m = Nothing
       | n < tag1 = indexJ n jl1
       | n >= tag1 = indexJ (n - tag1) jl2
       where tag1 = tag jl1