Haskell: Replacing element with a given key in an association list

Question

I've got to make a function which is given a key (as a String), a value (as a String) and an association list of keys and values (as [(String, String)]). The function is meant to add the key/value pair on to the end of the list, and, if the key is already present in the list with an associated value, delete the old value.

I've tried using lookup on the key and association list, but I'm not sure what to do with the output - the output type of the lookup function is Maybe String, and I can't seem to do list functions (like dropping elements) on it. Is there any way I can look through the list and delete any list element with a given key without knowing the value associated with it?

Answer 1

Here's a simple function that does what you want. It takes the new key value pair and puts that at the front of the given assoc list with that key filtered out.

addOrReplace :: Eq k => k -> v -> [(k, v)] -> [(k, v)]
addOrReplace key value assoc = (key,value):(filter ((key /=).fst) assoc)

The function fst is defined as:

fst (first,second) = first

filter takes a predicate and a list, and returns the list with only those elements that satisfy the predicate.

Edit: Split the key value pair in the parameters for addOrReplace as Tom suggests.

Answer 2

You should probably write a recursive function that takes the new key/value-pair and the existing list as parameters, and loops through the list to generate a new list with the new value inserted. For each list element you check if the key is the same as the one you want to insert. If it's different you keep that old element, if it's the same you add the new item instead of the old one. If you reach the end of the list without finding the key you just insert the new item there at the end.