fork download

-- Converts a list of key/values into a map, aggregating values
-- e.g. [(1, "one"), (2, "two"), (1, "uno")] -> Map {1: ["one", "uno"], 2: ["two"]} 
keyValuesToMap :: (Eq k, Ord k, Ord v) => [(k, v)] -> M.Map k [v]
keyValuesToMap = M.fromList . groupValues
    where
        groupValues :: (Eq k, Ord k, Ord v) => [(k, v)] -> [(k, [v])]
        groupValues = map foldValues . groupBy (\(x, _) (y, _) -> x == y) . sort 
 
        -- Combines values of the same key into a list
        -- e.g. [(1, 2), (1, 4), (1, 5)] -> (1, [2, 4, 5]) 
        -- Precondition: the key must be the same for all values in the input list
        foldValues :: (Eq k, Ord k, Ord v) => [(k, v)] -> (k, [v])
        foldValues xs = foldr accumulateSameKeyValues (key, []) xs
            where
                key = fst . head $ xs
 
                accumulateSameKeyValues :: (Eq k, Ord k, Ord v) => (k, v) -> (k, [v]) -> (k, [v])
                accumulateSameKeyValues (key1, value) (key2, values)
                    | key1 == key2 = (key1, value:values)
                    | otherwise = error "Cannot group values for different keys."

[1 of 1] Compiling Main             ( prog.hs, prog.o )

prog.hs:3:55: error:
    Not in scope: type constructor or class ‘M.Map’
    No module named ‘M’ is imported.
  |
3 | keyValuesToMap :: (Eq k, Ord k, Ord v) => [(k, v)] -> M.Map k [v]
  |                                                       ^^^^^

prog.hs:4:18: error:
    Not in scope: ‘M.fromList’
    No module named ‘M’ is imported.
  |
4 | keyValuesToMap = M.fromList . groupValues
  |                  ^^^^^^^^^^

Standard output is empty