CodinGame - MIME Type

mime.hs

import Control.Monad
import Data.Char
import Data.Maybe
import Data.List

main = do
    [n, q] <- replicateM 2 (fmap read getLine)
    types <- replicateM n (fmap ((\ [x, y] -> (map toLower x, y)) . words) getLine)
    replicateM q $ do
        ext <- fmap (extension . map toLower) getLine
        putStrLn (fromMaybe "UNKNOWN" (lookup ext types))
        
extension s =
    if '.' `elem` s
        then dropDot s
        else ""
    where
        dropDot s =
            case elemIndex '.' s of
                Just i -> dropDot (drop (i + 1) s)
                Nothing -> s