Garmelon/advent-of-code
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[hs] Clean up 2020_20

Browse source
This commit is contained in:
Joscha 2020-12-20 14:07:18 +00:00
parent 2a8f1cb9b2
commit 562c836b11
1 changed files with 10 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

32
hs/src/Aoc/Y2020/D20.hs
View file

@ -5,6 +5,7 @@ module Aoc.Y2020.D20
) where ) where
import Control.Monad import Control.Monad
import Data.Bool
import Data.List import Data.List
import Data.Maybe import Data.Maybe
@ -29,26 +30,17 @@ tTop (Tile l) = head l
tBottom :: Tile -> [Bool] tBottom :: Tile -> [Bool]
tBottom (Tile l) = last l tBottom (Tile l) = last l
tTranspose :: Tile -> Tile
tTranspose (Tile l) = Tile $ transpose l
tFlipV :: Tile -> Tile tFlipV :: Tile -> Tile
tFlipV (Tile l) = Tile $ reverse l tFlipV = Tile . reverse . unTile
tFlipH :: Tile -> Tile
tFlipH (Tile l) = Tile $ map reverse l
tTurnCw :: Tile -> Tile tTurnCw :: Tile -> Tile
tTurnCw = tFlipH . tTranspose tTurnCw = Tile . transpose . reverse . unTile
tTurnCcw :: Tile -> Tile
tTurnCcw = tFlipV . tTranspose
tRotations :: Tile -> [Tile] tRotations :: Tile -> [Tile]
tRotations = take 4 . iterate tTurnCw tRotations = take 4 . iterate tTurnCw
tVariations :: Tile -> [Tile] tVariations :: Tile -> [Tile]
tVariations t = tRotations t ++ tRotations (tFlipH t) tVariations t = tRotations t ++ tRotations (tFlipV t)
tShrink :: Tile -> Tile tShrink :: Tile -> Tile
tShrink (Tile l) = Tile $ tail $ init $ map (tail . init) l tShrink (Tile l) = Tile $ tail $ init $ map (tail . init) l
@ -146,20 +138,16 @@ solver tiles = do
let cornerIds = map (fst . (placed Map.!)) $ corners $ Map.keys placed let cornerIds = map (fst . (placed Map.!)) $ corners $ Map.keys placed
print $ product cornerIds print $ product cornerIds
putStrLn ""
putStrLn ">> Part 2" putStrLn ">> Part 2"
let bigTile = foldr1 tJoinV $ map (foldr1 tJoinH . map tShrink) $ layout placed let bigTile = foldr1 tJoinV $ map (foldr1 tJoinH . map tShrink) $ layout placed
monstersFound = maximum $ map (monsters . unTile) $ tVariations bigTile
-- Pretty printing
-- let (Tile l) = bigTile
-- for_ l $ \row -> do
-- for_ row $ \field -> putStr $ bool "." "#" field
-- putStrLn ""
let monstersFound = maximum $ map (monsters . unTile) $ tVariations bigTile
hashes = length $ filter id $ concat $ unTile bigTile hashes = length $ filter id $ concat $ unTile bigTile
print monstersFound
print hashes
print $ hashes - monstersFound * 15 print $ hashes - monstersFound * 15
-- Pretty-printing
putStrLn ""
putStr $ unlines $ map (map $ bool '.' '#') $ unTile bigTile
day :: Day day :: Day
day = dayParse parser solver day = dayParse parser solver