2023-03 in haskell

4 files changed, 300 insertions, 0 deletions

diff --git a/2023/03-Gear_Ratios/example b/2023/03-Gear_Ratios/example
new file mode 100644
index 0000000..b20187f
--- /dev/null
+++ b/2023/03-Gear_Ratios/example
@@ -0,0 +1,10 @@
+467..114..
+...*......
+..35..633.
+......#...
+617*......
+.....+.58.
+..592.....
+......755.
+...$.*....
+.664.598..
diff --git a/2023/03-Gear_Ratios/first.hs b/2023/03-Gear_Ratios/first.hs
new file mode 100644
index 0000000..9926363
--- /dev/null
+++ b/2023/03-Gear_Ratios/first.hs
@@ -0,0 +1,71 @@
+-- very slow with runghc, use ghc -O3 -o first first.hs
+-- requires cabal install --lib megaparsec parser-combinators
+module Main (main) where
+
+import Control.Applicative.Permutations
+import Control.Monad (void, when)
+import Data.Char qualified as C
+import Data.Either
+import Data.List qualified as L
+import Data.Map qualified as M
+import Data.Maybe
+import Data.Set qualified as S
+import Data.Void (Void)
+import Text.Megaparsec
+import Text.Megaparsec.Char
+
+exampleExpectedOutput = 4361
+
+data Number = Number Int Int Int Int deriving Show -- Value x1 x2 y1
+data Symbol = Symbol Int Int deriving (Eq, Ord, Show) -- x y
+data Input = Input [Number] [Symbol] deriving Show
+
+type Parser = Parsec Void String
+
+parseNumber :: Parser Number
+parseNumber = do
+  (SourcePos _ y' x1') <- getSourcePos
+  value <- read <$> some digitChar
+  (SourcePos _ _ x2') <- getSourcePos
+  let (x1, x2, y) = (unPos x1', unPos x2' - 1, unPos y')
+  return $ Number value x1 x2 y
+
+parseSymbol :: Parser Symbol
+parseSymbol = do
+  (SourcePos _ y' x') <- getSourcePos
+  void asciiChar
+  let (x, y) = (unPos x', unPos y')
+  return $ Symbol x y
+
+parseNumbersAndSymbols :: Parser (Either Number Symbol)
+parseNumbersAndSymbols = do
+  many $ void (char '.') <|> void eol
+  something <- (Left <$> parseNumber) <|> (Right <$> parseSymbol)
+  many $ void (char '.') <|> void eol
+  return something
+
+parseInput' :: Parser Input
+parseInput' = do
+  things <- some parseNumbersAndSymbols <* eof
+  return $ Input (lefts things) (rights things)
+
+parseInput :: String -> IO Input
+parseInput filename = do
+  input <- readFile filename
+  case runParser parseInput' filename input of
+    Left bundle -> error $ errorBundlePretty bundle
+    Right input' -> return input'
+
+compute :: Input -> Int
+compute (Input nums syms) = sum . map (\(Number v _ _ _) -> v) $ L.filter process nums
+  where
+    process :: Number -> Bool
+    process (Number _ x1 x2 y) = not . S.null $ S.fromList [Symbol x y|x<-[x1-1..x2+1],y<-[y-1..y+1]] `S.intersection` S.fromList syms
+
+main :: IO ()
+main = do
+  example <- parseInput "example"
+  let exampleOutput = compute example
+  when  (exampleOutput /= exampleExpectedOutput)  (error $ "example failed: got " ++ show exampleOutput ++ " instead of " ++ show exampleExpectedOutput)
+  input <- parseInput "input"
+  print $ compute input
diff --git a/2023/03-Gear_Ratios/input b/2023/03-Gear_Ratios/input
new file mode 100644
index 0000000..15cb53f
--- /dev/null
+++ b/2023/03-Gear_Ratios/input
@@ -0,0 +1,140 @@
+....401.............425.......323......791......697...............963............................................420........................
+...*..................................%......#.....*....290.........................492.............656...@953.....................+830.....
+..159...........823...33.717.....572.......806...896......-.....335....834......815.............791....*..............776...................
+.........-.....#........*.........*..................715..........*.....*........*.....................5...*.....................688........
+....=..=..573..............212.......553.632....622....*.......260....29.........692.129....727...........290.........%32.....%...*..441....
+.791..998..............................*............531.......................84.......*..../.......506%......=............240..831..*......
+.....................321.586........447...&.-276.21......@..........12.......&.......-..333...................619..94................982....
+..........@...........#..*..............525.......*...183............*............407.......23.........%............&......846..............
+....97..686.............89.....................-..494..............875....*607.............*.......239.407..................%.......529.....
+.....+.........815..703............208.=555.856...........@............133........................*..........672...501........898#...*......
+...............*.......*...659.......*....................183..............528.......614.......4....................................39......
+...602..........804..313....*.......480..............602.............354......*877................832@.....140...$...........707.......222..
+...*....999.................23.264..............516.....*.......-....*....................................*.....722......24....*........*...
+436....+................./......./..........765*......37.....843....218..........852*68....290............410...........*.....539....896....
+.........../...#........907............587...............&................149...............*.......*..............#780..707................
+........349....603.598....................-....203.....178.......431......./...............152......571....926.750.............127..........
+.............@........*176.283...................*...%......642......../......20..............................*.........@.............534...
+...749..../.194..............*.......337.395*....682.429.=...#..........186.....*.........682..../......498..........%.8.............*......
+....*..324..........893*...878......*........659..........65....671.189.......641...........+.859.........$.......149....648........53.+....
+...776......$.....................667.................432...........*..............697............282.901...................*889........471.
+............103............../........-..................-...748&.257.............&...........739*....*......491*543................997.....
+.........2.........813*.....278..316.333.....296.142.+...................673.%....../.....+.........490...................790........*......
+......@.*.......................*...............*....643.464............*....25....108..807..179.........931.......289..........365...522...
+....367.250.927../.............754.........14...............*.+..........23...................*......&.....*..%..........323................
+................939...644..............791..*.551....35...937..806...............&.............536.930...703...565......*...............*640
+534$.................*.........$..448.....=......*...................707*658.....33....920..74.....................844................36....
+.......572....672...875......770.+................380.............52......................*.......-....251............%........*909......677
+.......*.......=........................658.889.........882.....=......................534..*..291...%....*.................289.............
+.212......993..............438./..........*....@..532......*586.163....213...307...........167.....981.700..................................
+....*.....*..................-..12...611...196.......*309.....................+.......*685............................403....243..+385......
+59...428..14...22.747....990............*........575...............................247.....985....&..................*.......*..............
+.................*........%...539.......416.712.....*2..408.......%....................999.......275......738.....715......670..............
+..357........653..............*....374..........285................785.........384......*.............277*........................765.137...
+...@..........*..@632......361.......=..............35.........329..............................=858..............798.................*.....
+......................216*.......459..................*..........&........479.....255......302......................*.530.........269..453..
+............./.573........153.....*........#.......462...%..........45......+.122*............*672....545........157.....*862.....*.........
+820....%...977.+...................726...790.199.........631....308..................755..............*...460..................643....#237..
+.......229.............727..425............../..........................228........#..$...............385....&...........695................
+...................978.*....*....................700....*.........256..@.........625......311.170...+................642*...................
+.492.....983.........-.577.743..................*....267.521........%....479.......................696.980......................*...........
+....*820.#....................................236.................*......*...856.............137...........489......409........177..450.....
+974.........479.$......182..456......./.............*755..617..333.849.........*...............*..........*..........=...-...........*......
+...+..........*..83.#....@.=...........893.......610.........*.................727....6*..488.839...449..541../97......560.........449......
+............208......168......971..........................547..........559............................=...........237...................641
+.......336...................*....563..993.911.=......853................*...............391...@572.................*.......*......614..*...
+........$...........579.....810....-......*....480....*......649.138..410..201..604........*..............450#.....485...124.841..*......790
+..399...........928.............67..................#..295......*............*...#....333...83...#111.............................635.......
+...=........./.....*..807...529..*......642#......603......704.........@....599......&.................=....................................
+..............233.618..../........114........566......100..*.....423.26...............................168................555$..........#625.
+.....608*372..................543..........*.*............4.....*.......+374.......436*...................680.135....................*......
+.................................@...385.309..111......69....991..127........693#......985....25..856*96.+.......*860..........%..278.160...
+.......351.......495=.772..............*...............*...........*...........................@............569...............905...........
+........................*.$402........640.*.............366.........350...623&...250.............446*341.....=.........................292..
+....#..*...595.289...347..........75.......844......307..........................*....585.....*..................766+...................*...
+.148..906.....*..........232.449...*...........................................326..$...-..102.595.....406............-...........650....682
+..........619...23........$...*.....138.893.......890...79....53.........118........966................+..............233............*......
+.........*......../..........62..........*........*....*.............@..*....................245...........-....806.........98........318...
+....534+.264..993...420.................675..77..659..263..........615.405...............=..*......%..75...614.*..................4.........
+...............*..............421.........................834*287.......................813..651.902...*........134.....$...........334.....
+....631....=..443........................352..........941...............317.......*286...............562.855............690....387.....*....
+.27...$.430......................89....-....*90...&......%..........155.......899.......%59....*986.................775..............25.....
+...*....................-.....+...*...952.........62.........478...........&...../..........694.........775.....640..$......................
+....102..........+197....109.270.453.....................................641.......123..........573....*.......=............880.403+........
+.........87.186..........................709.+219........224.726...882.............*...868..203*....732............928*607..*...............
+..819....*..*.............*....81........*....................-../..%...+..........700.=.......................149............*....438..=...
+....#.875..223.106.........646......%.....519...579.............38.....883.626@....................149..........=..........205.272./....480.
+.......................965.........332.........*......221.292...........................594..945......*931..365....814*214..................
+..228*....29............*....79...........+.....811.....*..*..592..........681.............*....#.427........*................170..58.......
+......563...*........817..93*...+60..507...898........796.942...*..=335.......@...490....488.......-......974....817...........*....+.......
+...........519..&...................*..........948.............454......184.........*.........=.......................409......947..........
+...............272..........257....712............+.@.....338..............*381...824..923..756.............971*181....=....+......430......
+....228.............709.......&......................685.....*..597....135......+.....*..........+......471..................452..*......923
+....*..................#...........271.....................167.*.........*.......325.935..........224......*.....749&...-..........587......
+..464.../199..$..........50...........*..794..........36.......733....................................@....414...........605................
+...............765.%.......=.215*717........*715....................210......845.......................643........66...........*908.........
+.........215........761............................544....781...390*.....576..*....869..781....517................*.....356.664.......=.....
+..............316...........235/...............715*.....-....*.............*..130..........*..@..........&.........564.*...............89...
+...................................&....169...........353....390....57.71.847...........716.........466..516..651......447..................
+.......459..................101.379.....*......324...................#.#............943..............*...........%.........$625.../...586...
+........*.....173...................198..306...*....313........51............629...*..........-958..228.178.......................967.......
+......714...@.........206.......934*.........109...........*.........../......*....832...................*........934.....921..........897..
+...........487............700......................*507..415....349%...179...961................437.......489.......*............../...*....
+.................*440.952*................667...533.....................................359.618....*927.........................968...412...
+...........-..778..................667.....@..........764..401.........279.......#.748.....*................534*.....444....................
+...835..779......../.....443.........*.........................53/......=......774......=....784...204..385.....929..........+...861........
+......%..........74..947*.....187&.874..........&.........................545........752.......$....$...*..................308.....*..284...
+........#.................190...........445$..784...51*226...................*............*245..........435.......=...200..........88.*.....
+640.....310...&...........*......296..........................................107......943..........244...........662......=..440+....302...
+...*..........67........381..............856....349...........790*...911........................901...*......733........844.................
+.327..............885........169.....419........*.....62..........70..@..664...........=...422%...*....617.....*....................230*....
+...........916...@.............................95....@.....+.........................355.........104...........149......................569.
+.............*.........................230.../..............350..............................432.....710............=.........=.............
+..925..545..634.....@..297*933...........*....839...33...........34.....162.......577..248.....#...........814...524..........790......9/...
+...*...............66....................421............606+..*../.......&.........*....&..129..............=......................529......
+659..........253...........$504.......*......................379....&........./..860......+..............&...............%.....658*....310..
+........208...*........................354..........................647......326...............911.....503.......11.......293...............
+....*.....&.617..511.........885.906......................166/......................589..........*..........749....*931..........915........
+.852.908........................*................../........................297.470*........#628..269...613......#................*.........
+.........................716.................689...460...737*502...510.............../...........................778.......373..734.....+...
+..520=.........../.....*....*226........326.....+.................+............971...362........800$..904.303.........*874..........=..548..
+............554.889.720.854......808.....*..258............284*........639/...*...........................*........603.............839......
+...........*........................*690.71..*.........314.....795.86.........53.................786%...134...................890*..........
+663.....225...%........513...................951..........*........*...............275.995...764.....................=............30........
+...*...........353....*...........%......................468.....87..225........31..=.....%.=......../.......886.....84.....................
+..694......166........243........380...275..287..............261.....@.....888....*............162...278.......&..........351.534........197
+........$.......209.-......261...........*.......876.....15.....*.........*........712.....870...*.......897.....46...........#.............
+......450.........%..463.....%.679....119.....=.*....@...@.........871...408..995*...........*...922.+..........*...66.466..................
+..............803.........................$.250...493...............&.............356.519.552........737.......153....*.............+.......
+......982........*.....................540...................641.......................*......&....................................851../...
+..............638..............=.306....................+938..@................*520....511.174......*..........-......*503...832.........219
+.292..239...................775..*......334..../582...................173...859..................238.678......117..429.......%...708.534....
+........*........240....491.....546...................+................*.............%....498.............................55........*.....74
+.816.....406.......*...&..............181...........765.......776....+..636...&337...264..*............*716..$908...........*...............
+...................860....................980...................$..323....................163.......851............*.......396......501.....
+...-..20*197................722..........*...............................&.....408..............109.......794*155..590........./....$.......
+.482...........257..........*............999./850.........482..........933........*.*982...../.....*..........................826.......908.
+.......%597......@..345.329..172.....295...........349....*.....=527........174.727........645...167......@..........*......................
+.....................*.....+..........$.............*....725...............*............$............+.447..723...346.759............967....
+............638..............................589.....................................=..312..+....951..........=..........809*..........#...
+.............@............325..........322.........&......................508......359......768........8..&.........992.....................
+.783.32...................*............*.........656.............98*178.....%.....................74*.....2...410&....@............+........
+...*...*..54*525...112..720...66......641.+280.......................................................467.......................580..348.....
+..699.673......................................491..869............#......171..............98............473......%..............$..........
+........................%.....214.......303$......#..%.......191.852..560*.......835..550...$.150.........#........291.......240............
+.......579...982.........713....*...536.......................+...................*....#........$..840.......43................*............
+......$.........*....*.......606......+....48...........993.......838*744.880....969........@........*......*....*726..........168...&......
+...............395.579............340......*........386.=...420..............*........833..835........257...842....................810...635
+......#......................%...........983...632.@........*........-...%..157...193*...........................@..137....209..............
+.....819.....................278..377........................578...195..230........................$......586.985...............263.417.....
+................592....../............*......../.............................922.......350........138.156*............698...78*.........$571
+....................405/..573.764....257.@940...34.977$..441....304.............*.....*................................*.......93..600......
+...643....675...262..........................................................993.....556...........314........278.258.322............*......
+...*.........*.....&................/.....................222.........959................@..121.......*...930.......#.......170*462.....588.
+384....+..19.863..............407=.283.........972...340....#....427*....*737...........558...%.....91......*...........................=...
+.....812...@.........298..563....................*......*............506..........845......................634...&............624+...&......
+.....................*.....=.....*.............492.563...950..827*...............+....................968......652..................49......
+.730.........498..382.........399.......824.........&...................390.630........248...........*.......-......448......-..............
+................*......333..........832*.....#...............$...565.....*...............*............701..564.........*.115..718.......313.
+.960...*965....993......*.....*..............604..........730...*..................234&..662.....+..............=69.930...*.......967*......
+.....30..............355...978...........67.........419........239..........732...............645..........................484........664...
diff --git a/2023/03-Gear_Ratios/second.hs b/2023/03-Gear_Ratios/second.hs
new file mode 100644
index 0000000..d02cb31
--- /dev/null
+++ b/2023/03-Gear_Ratios/second.hs
@@ -0,0 +1,79 @@
+-- very slow with runghc, use ghc -O3 -o second second.hs
+-- requires cabal install --lib megaparsec parser-combinators
+module Main (main) where
+
+import Control.Applicative.Permutations
+import Control.Monad (void, when)
+import Data.Char qualified as C
+import Data.Either
+import Data.List qualified as L
+import Data.Map qualified as M
+import Data.Maybe
+import Data.Set qualified as S
+import Data.Void (Void)
+import Text.Megaparsec
+import Text.Megaparsec.Char
+
+exampleExpectedOutput = 467835
+
+data Number = Number Int Int Int Int deriving Show -- Value x1 x2 y1
+data Symbol = Symbol Int Int deriving (Eq, Ord, Show) -- x y
+data Input = Input [Number] [Symbol] deriving Show
+
+type Parser = Parsec Void String
+
+parseNumber :: Parser Number
+parseNumber = do
+  (SourcePos _ y' x1') <- getSourcePos
+  value <- read <$> some digitChar
+  (SourcePos _ _ x2') <- getSourcePos
+  let (x1, x2, y) = (unPos x1', unPos x2' - 1, unPos y')
+  return $ Number value x1 x2 y
+
+parseSymbol :: Parser Symbol
+parseSymbol = do
+  (SourcePos _ y' x') <- getSourcePos
+  void $ char '*'
+  let (x, y) = (unPos x', unPos y')
+  return $ Symbol x y
+
+parseNumbersAndSymbols :: Parser (Either Number Symbol)
+parseNumbersAndSymbols = do
+  many $ void (char '.' <|> symbolChar <|> char '#' <|> char '%' <|> char '@' <|> char '-' <|> char '/' <|> char '&') <|> void eol
+  something <- (Left <$> parseNumber) <|> (Right <$> parseSymbol)
+  many $ void (char '.') <|> void eol
+  return something
+
+parseInput' :: Parser Input
+parseInput' = do
+  things <- some parseNumbersAndSymbols <* eof
+  return $ Input (lefts things) (rights things)
+
+parseInput :: String -> IO Input
+parseInput filename = do
+  input <- readFile filename
+  case runParser parseInput' filename input of
+    Left bundle -> error $ errorBundlePretty bundle
+    Right input' -> return input'
+
+compute :: Input -> Int
+compute (Input nums syms) = sum $ map process syms
+  where
+    process :: Symbol -> Int
+    process sym = case L.foldl' process' [] nums of
+      [a, b] -> a * b
+      _ -> 0
+      where
+        process' :: [Int] -> Number -> [Int]
+        process' acc (Number v x1 x2 y') | not . S.null $ S.fromList [Symbol x y|x<-[x1-1..x2+1],y<-[y'-1..y'+1]] `S.intersection` S.singleton sym = v:acc
+                                         | otherwise = acc
+
+main :: IO ()
+main = do
+  example <- parseInput "example"
+  print example
+  let exampleOutput = compute example
+  when  (exampleOutput /= exampleExpectedOutput)  (error $ "example failed: got " ++ show exampleOutput ++ " instead of " ++ show exampleExpectedOutput)
+  input <- parseInput "input"
+  print $ compute input
+