natty
/
aoc23
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Slightly less ugly part 2 solution for day 02

This commit is contained in:
Natty 2023-12-02 15:01:55 +01:00
parent c7ae8ad496
commit 8501f5e48b
Signed by: natty
GPG Key ID: BF6CB659ADEE60EC
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
day02b/src/Main.hs
View File

@ -22,10 +22,13 @@ mapCubes :: T.Text -> (Integer, [[(Integer, String)]])
mapCubes line = parseParts $ T.splitOn (T.pack ": ") line
mergeMaxCube :: [(Integer, String)] -> (Integer, String) -> [(Integer, String)]
mergeMaxCube acc (aa, col) = (\(a, c) -> if c == col then (max a aa, c) else (a, c)) <$> acc
mergeMaxCube (h@(ax, colx) : xs) b@(ab, col)
| colx == col = (max ax ab, colx) : xs
| otherwise = h : mergeMaxCube xs b
mergeMaxCube [] v = [v]
main :: IO ()
main = do
lns <- lines <$> getContents
let values = mapCubes . T.pack <$> lns
print $ sum $ product . (fst <$>) . foldl mergeMaxCube [(0, "red"), (0, "green"), (0, "blue")] . concat . snd <$> values
print $ sum $ product . (fst <$>) . foldl mergeMaxCube [] . concat . snd <$> values