blob: d7006acece762f46f6395dd6eb1fbc307a97fef6 (
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
module Day21A where
import System.Environment (getArgs)
import Data.List (nub)
import Data.List.Split (splitOn)
-- Types
type Food = ([String], [String])
-- Parse a single line
parseFood :: String -> Food
parseFood x = (splitOn " " ings, splitOn ", " algsNParen)
where [ings, algs] = splitOn " (contains " x
algsNParen = take (length algs - 1) algs
-- Parse a whole file
parseFoods :: String -> [Food]
parseFoods = map parseFood . lines
parseFile :: String -> IO [Food]
parseFile f = do
c <- readFile f
return (parseFoods c);
-- Returns true if x is a sublist of y.
sublist :: Eq a => [a] -> [a] -> Bool
sublist xs l = foldr (\x a -> a && x `elem` l) True xs
-- Get a list of foods with the given allergen
withAls :: [Food] -> String -> [Food]
withAls fs x = filter ((x `elem`) . snd) fs
-- Get a list of foods with the given ing
withIng :: [Food] -> String -> [Food]
withIng fs x = filter ((x `elem`) . fst) fs
-- Get all ingredients in all the given foods
allIngs :: [Food] -> [String]
allIngs fs = nub $ concatMap fst fs
-- Get all allergens in all the given foods
allAls :: [Food] -> [String]
allAls fs = nub $ concatMap snd fs
-- Get a list of foods that are in all foods with the given allergen
getSynonymous :: [Food] -> String -> [String]
getSynonymous fs x = filter ((xs `sublist`) . withIng fs) (allIngs fs)
where xs = withAls fs x
-- Usage: runghc --ghc-arg="-package split" 21a.hs inputs/day21
main :: IO ()
main = do
args <- getArgs;
fs <- parseFile $ head args;
let syn = nub $ concatMap (getSynonymous fs) (allAls fs);
let notSyn = filter (not . (`elem` syn)) (allIngs fs);
print $ sum $ map (length . withIng fs) notSyn;
return ();
|
