From c6075f736e67e47b940abb9c1a0c54753d4b55b8 Mon Sep 17 00:00:00 2001
From: Orangerot <purple@orangerot.dev>
Date: Fri, 17 May 2024 15:41:55 +0200
Subject: AoC 2023

---
 2023/day03/Main.hs | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 2023/day03/Main.hs

(limited to '2023/day03/Main.hs')

diff --git a/2023/day03/Main.hs b/2023/day03/Main.hs
new file mode 100644
index 0000000..d9d8f4b
--- /dev/null
+++ b/2023/day03/Main.hs
@@ -0,0 +1,28 @@
+import Data.Char
+import Data.List
+
+-- findBegin :: ((Int, Int), Char) -> [((Int, Int), Char)] -> ((Int, Int), Char)
+findBegin s@((x,y), c) dict = case filter (\(p,_) -> p == (x,y-1)) dict of
+  [] -> s
+  n:xs -> findBegin n dict
+
+main :: IO ()
+main = do
+  inputLines <- lines <$> getContents
+
+  let distances = concat (map (\x -> (map (\y -> (x,y)) [-1..1])) [-1..1])
+
+  let numInputLines = zip [0..] (map (zip [0..]) inputLines)
+  let charCoords = concat (map (\(x, l) -> (map (\(y, c) -> ((x,y),c)) l)) numInputLines)
+  let symbols = map (\(p, c) -> p) (filter (\(_,c) -> not (c == '.' || isDigit c)) charCoords)
+  let digits = filter (\(_,c) -> isDigit c) charCoords
+  let serialChars = filter (\((x,y),_) -> any (\x -> elem x symbols) (map (\(x1,y2) -> (x + x1, y + y2)) distances)) digits
+  let serialBegins = nub $ map (\x -> findBegin x digits) serialChars
+  let serialNumbers = map (\((x,y), c) -> read (takeWhile isDigit (drop y (inputLines!!x))) :: Integer) serialBegins
+  -- print charCoords 
+  -- print symbols
+  print (sum serialNumbers)
+  -- convert input to [((x,y), Char)]
+
+  -- mapM_ putStrLn inputLines
+
-- 
cgit v1.2.3