Brool brool (n.) : a low roar; a deep murmur or humming
State Monads in Haskell
I would like Haskell better if it didn’t do its best to make me feel stupid. Tasks that are easy in other languages, such, say, maintaining some state or generating a random number, become difficult. After banging my head on the State monad I finally arrived at a kind of understanding, which I put here so I can restore context in the future!
Presume that you’ve got a function that, given a list, returns a list of booleans corresponding to whether the value was the first occurrence in the list. In Python it would look something like this:
def is_first(lst):
found = set()
r = []
for item in lst:
r.append(item not in found)
found.add(item)
return r
For example:
>>> is_first([1,2,3,1,4])
[True, True, True, False, True]
The most simplistic way of converting this to Haskell would require using recursion for the looping, resulting in the following (non-tail-recursive) function:
first lst = first' lst []
first' [] _ = []
first' (h:t) state =
if h `elem` found then False:(first' t state)
else True:(first' t (h:state))
The first’ function needs to “carry around” state as it loops around. Now, say that the lookup function is more complicated than we thought, so we break it out into a separate function. We have to pass around the state to all functions that access it, resulting in:
first lst = first' lst []
first' [] _ = []
first' (h:t) state =
if (isfound h state) then False:(first' t state)
else True:(first' t (h:state))
isfound h state = h `elem` state
Since isfound references the state, then it needs to be passed in explicitly.
I think that my main problem with understanding the State monad was that I kept wanting to treat the state as a global variable; so, I was thinking of “s <- get” and “put s” as references to a global stored somewhere, which left me with the question of “where is the initial put that sets the state?” Instead, though, you should think of “s <- get” and “put s” as references to an implicitly passed parameter.
The State Monad allows this particular pattern to be changed so that the state doesn’t need to be explicitly passed. Rewriting:
state_first lst = evalState (state_first' lst) []
state_first' [] = return []
state_first' (h:t) = do state <- get
let found = h `elem` state
put (if found then state else (h:state))
rest <- state_first' t
return $ (not found):rest
Basically, given a Haskell function, you can convert it into the state monad by:
- Removing the explicitly passed state
- Changing all references to the state parameter to “param <- get”
- Instead of recursively descending, use “put new-state-value”
- Calling a state returns using “<-” to ‘strip’ the value
Now, this doesn’t seem more elegant, but note that if you have to break functionality out, the state parameter no longer needs to be explicitly passed:
state_first lst = evalState (state_first' lst) []
state_first' [] = return []
state_first' (h:t) = do state <- get
found <- isfound h
put (if found then state else (h:state))
rest <- state_first' t
return $ (not found):rest
isfound h = do state <- get
return $ h `elem` state
UPDATED: following a question on #haskell (which was very friendly, especially to a complete newcomer), the following more elegant ways of solving this came up:
-- using mapAccum
first lst = snd $ mapAccumL first' [] lst
where first' acc x = (if found then acc else x:acc, not found)
where found = x `elem` acc
-- with mapM (not exactly equivalent to example)
first lst = evalState first' []
where first' = mapM (\a -> do state <- get; put (a:state); return $ not (a `elem` state)) lst
-- with mapM, equivalent to example
first lst = evalState first' []
where first' = mapM isfound lst
isfound a = do state <- get
let found = a `elem` state
put (if found then state else (a:state))
return $ not found
For more information:
Roll Your Own IRC
Bot
Grok Haskell Monad
Transformers
Monads For The Working Haskell
Programmer
Haskell Monads: Another
View
Discussion
Comments are moderated whenever I remember that I have a blog.
