diff --git a/day-08-haskell/.gitignore b/day-08-haskell/.gitignore new file mode 100644 index 0000000..c368d45 --- /dev/null +++ b/day-08-haskell/.gitignore @@ -0,0 +1,2 @@ +.stack-work/ +*~ \ No newline at end of file diff --git a/day-08-haskell/Setup.hs b/day-08-haskell/Setup.hs new file mode 100644 index 0000000..9a994af --- /dev/null +++ b/day-08-haskell/Setup.hs @@ -0,0 +1,2 @@ +import Distribution.Simple +main = defaultMain diff --git a/day-08-haskell/app/Main.hs b/day-08-haskell/app/Main.hs new file mode 100644 index 0000000..4a66fd4 --- /dev/null +++ b/day-08-haskell/app/Main.hs @@ -0,0 +1,17 @@ +module Main (main) where + +import System.Environment.Blank ( getArgs ) +import Task1 (task1) +import Task2 (task2) + +main :: IO () +main = do + args <- getArgs + case args of + ["1", file] -> do + input <- readFile file + print $ task1 input + ["2", file] -> do + input <- readFile file + print $ task2 input + _ -> error "Usage: <1|2> " diff --git a/day-08-haskell/day08-haskell.cabal b/day-08-haskell/day08-haskell.cabal new file mode 100644 index 0000000..3cacd12 --- /dev/null +++ b/day-08-haskell/day08-haskell.cabal @@ -0,0 +1,53 @@ +cabal-version: 2.2 + +-- This file has been generated from package.yaml by hpack version 0.37.0. +-- +-- see: https://github.com/sol/hpack + +name: day08-haskell +version: 0.1.0.0 +homepage: https://github.com/githubuser/day08-haskell#readme +bug-reports: https://github.com/githubuser/day08-haskell/issues +author: Author name here +maintainer: example@example.com +copyright: 2024 Author name here +license: BSD-3-Clause +build-type: Simple + +source-repository head + type: git + location: https://github.com/githubuser/day08-haskell + +library + exposed-modules: + Common + Task1 + Task2 + hs-source-dirs: + src + ghc-options: -Wall -Wcompat -Widentities -Wincomplete-record-updates -Wincomplete-uni-patterns -Wmissing-export-lists -Wmissing-home-modules -Wpartial-fields -Wredundant-constraints + build-depends: + base >=4.7 && <5 + , containers + default-language: Haskell2010 + +executable day08-haskell-exe + main-is: Main.hs + hs-source-dirs: + app + ghc-options: -Wall -Wcompat -Widentities -Wincomplete-record-updates -Wincomplete-uni-patterns -Wmissing-export-lists -Wmissing-home-modules -Wpartial-fields -Wredundant-constraints -threaded -rtsopts -with-rtsopts=-N + build-depends: + base >=4.7 && <5 + , day08-haskell + default-language: Haskell2010 + +test-suite day08-haskell-test + type: exitcode-stdio-1.0 + main-is: Spec.hs + hs-source-dirs: + test + ghc-options: -Wall -Wcompat -Widentities -Wincomplete-record-updates -Wincomplete-uni-patterns -Wmissing-export-lists -Wmissing-home-modules -Wpartial-fields -Wredundant-constraints -threaded -rtsopts -with-rtsopts=-N + build-depends: + base >=4.7 && <5 + , day08-haskell + default-language: Haskell2010 diff --git a/day-08-haskell/package.yaml b/day-08-haskell/package.yaml new file mode 100644 index 0000000..914d5b6 --- /dev/null +++ b/day-08-haskell/package.yaml @@ -0,0 +1,49 @@ +name: day08-haskell +version: 0.1.0.0 +github: "githubuser/day08-haskell" +license: BSD-3-Clause +author: "Author name here" +maintainer: "example@example.com" +copyright: "2024 Author name here" +spec-version: 0.36.0 + +dependencies: +- base >= 4.7 && < 5 + +ghc-options: +- -Wall +- -Wcompat +- -Widentities +- -Wincomplete-record-updates +- -Wincomplete-uni-patterns +- -Wmissing-export-lists +- -Wmissing-home-modules +- -Wpartial-fields +- -Wredundant-constraints + +library: + source-dirs: src + dependencies: + - containers + +executables: + day08-haskell-exe: + main: Main.hs + source-dirs: app + ghc-options: + - -threaded + - -rtsopts + - -with-rtsopts=-N + dependencies: + - day08-haskell + +tests: + day08-haskell-test: + main: Spec.hs + source-dirs: test + ghc-options: + - -threaded + - -rtsopts + - -with-rtsopts=-N + dependencies: + - day08-haskell diff --git a/day-08-haskell/src/Common.hs b/day-08-haskell/src/Common.hs new file mode 100644 index 0000000..bb93bee --- /dev/null +++ b/day-08-haskell/src/Common.hs @@ -0,0 +1,37 @@ +module Common where + +import Data.Foldable (toList) +import qualified Data.Map.Strict as Map + +multiMapFromList :: Ord k => [(k, v)] -> Map.Map k [v] +multiMapFromList = Map.fromListWith (++) . map (\(k, v) -> (k, [v])) + +pairPermutations :: [a] -> [(a, a)] +pairPermutations [] = [] +pairPermutations [_] = [] +pairPermutations [x, y] = pure (x, y) +pairPermutations (x:xs) = map (\y -> (x, y)) xs ++ pairPermutations xs + +type Position = (Int, Int) +type Antenna = (Char, Position) +type World = (Position, [Antenna]) + +enumerate :: [a] -> [(Int, a)] +enumerate = zip [0..] + +readWorld :: String -> World +readWorld input = (determineSize $ lines input, readAntennas input) + where + determineSize :: [String] -> Position + determineSize ls = (maximum $ map length ls, length ls) + +readAntennas :: String -> [Antenna] +readAntennas input = enumerate (lines input) >>= readLine + where + readLine :: (Int, String) -> [Antenna] + readLine (y, line) = enumerate line >>= toList . readChar + where + readChar :: (Int, Char) -> Maybe Antenna + readChar (_, '.') = Nothing + readChar (x, char) = Just (char, (x, y)) + diff --git a/day-08-haskell/src/Task1.hs b/day-08-haskell/src/Task1.hs new file mode 100644 index 0000000..57a5a97 --- /dev/null +++ b/day-08-haskell/src/Task1.hs @@ -0,0 +1,28 @@ +module Task1 (task1) where + +import Common (Position, readWorld, multiMapFromList, pairPermutations) +import qualified Data.Map.Strict as Map +import qualified Data.Set as Set +import Control.Monad (join) + +task1 :: String -> Int +task1 input = case readWorld input of + ((width, height), antennas) -> length $ Set.fromList $ filter isInWorld $ join $ map antinodesOfFrequency $ Map.elems $ multiMapFromList antennas + where + antinodesOfFrequency :: [Position] -> [Position] + antinodesOfFrequency nodes = pairPermutations nodes >>= antinodesOfPair + + antinodesOfPair :: (Position, Position) -> [Position] + antinodesOfPair ((ax, ay), (bx, by)) = + let diffX = bx - ax + diffY = by - ay + in [(ax - diffX, ay - diffY), (bx + diffX, by + diffY)] + + isInWorld :: Position -> Bool + isInWorld (x, y) + | x < 0 = False + | x >= width = False + | y < 0 = False + | y >= height = False + | otherwise = True + diff --git a/day-08-haskell/src/Task2.hs b/day-08-haskell/src/Task2.hs new file mode 100644 index 0000000..461a500 --- /dev/null +++ b/day-08-haskell/src/Task2.hs @@ -0,0 +1,30 @@ +module Task2 (task2) where + +import Common (Position, readWorld, multiMapFromList, pairPermutations) +import qualified Data.Map.Strict as Map +import qualified Data.Set as Set +import Control.Monad (join) + +task2 :: String -> Int +task2 input = case readWorld input of + ((width, height), antennas) -> length $ Set.fromList $ join $ map antinodesOfFrequency $ Map.elems $ multiMapFromList antennas + where + antinodesOfFrequency :: [Position] -> [Position] + antinodesOfFrequency nodes = pairPermutations nodes >>= antinodesOfPair + + antinodesOfPair :: (Position, Position) -> [Position] + antinodesOfPair (a@(ax, ay), b@(bx, by)) = + let diffX = bx - ax + diffY = by - ay + in castInWorld a (-diffX, -diffY) ++ castInWorld b (diffX, diffY) + + castInWorld (sx, sy) (ox, oy) = + takeWhile isInWorld $ map (\f -> (sx + ox * f, sy + oy * f)) [0..] + + isInWorld :: Position -> Bool + isInWorld (x, y) + | x < 0 = False + | x >= width = False + | y < 0 = False + | y >= height = False + | otherwise = True diff --git a/day-08-haskell/stack.yaml b/day-08-haskell/stack.yaml new file mode 100644 index 0000000..d8656a4 --- /dev/null +++ b/day-08-haskell/stack.yaml @@ -0,0 +1,67 @@ +# This file was automatically generated by 'stack init' +# +# Some commonly used options have been documented as comments in this file. +# For advanced use and comprehensive documentation of the format, please see: +# https://docs.haskellstack.org/en/stable/yaml_configuration/ + +# A 'specific' Stackage snapshot or a compiler version. +# A snapshot resolver dictates the compiler version and the set of packages +# to be used for project dependencies. For example: +# +# snapshot: lts-22.28 +# snapshot: nightly-2024-07-05 +# snapshot: ghc-9.6.6 +# +# The location of a snapshot can be provided as a file or url. Stack assumes +# a snapshot provided as a file might change, whereas a url resource does not. +# +# snapshot: ./custom-snapshot.yaml +# snapshot: https://example.com/snapshots/2024-01-01.yaml +snapshot: + url: https://raw.githubusercontent.com/commercialhaskell/stackage-snapshots/master/lts/22/43.yaml + +# User packages to be built. +# Various formats can be used as shown in the example below. +# +# packages: +# - some-directory +# - https://example.com/foo/bar/baz-0.0.2.tar.gz +# subdirs: +# - auto-update +# - wai +packages: +- . +# Dependency packages to be pulled from upstream that are not in the snapshot. +# These entries can reference officially published versions as well as +# forks / in-progress versions pinned to a git hash. For example: +# +# extra-deps: +# - acme-missiles-0.3 +# - git: https://github.com/commercialhaskell/stack.git +# commit: e7b331f14bcffb8367cd58fbfc8b40ec7642100a +# +# extra-deps: [] + +# Override default flag values for project packages and extra-deps +# flags: {} + +# Extra package databases containing global packages +# extra-package-dbs: [] + +# Control whether we use the GHC we find on the path +# system-ghc: true +# +# Require a specific version of Stack, using version ranges +# require-stack-version: -any # Default +# require-stack-version: ">=3.1" +# +# Override the architecture used by Stack, especially useful on Windows +# arch: i386 +# arch: x86_64 +# +# Extra directories used by Stack for building +# extra-include-dirs: [/path/to/dir] +# extra-lib-dirs: [/path/to/dir] +# +# Allow a newer minor version of GHC than the snapshot specifies +# compiler-check: newer-minor diff --git a/day-08-haskell/stack.yaml.lock b/day-08-haskell/stack.yaml.lock new file mode 100644 index 0000000..e542442 --- /dev/null +++ b/day-08-haskell/stack.yaml.lock @@ -0,0 +1,13 @@ +# This file was autogenerated by Stack. +# You should not edit this file by hand. +# For more information, please see the documentation at: +# https://docs.haskellstack.org/en/stable/lock_files + +packages: [] +snapshots: +- completed: + sha256: 08bd13ce621b41a8f5e51456b38d5b46d7783ce114a50ab604d6bbab0d002146 + size: 720271 + url: https://raw.githubusercontent.com/commercialhaskell/stackage-snapshots/master/lts/22/43.yaml + original: + url: https://raw.githubusercontent.com/commercialhaskell/stackage-snapshots/master/lts/22/43.yaml diff --git a/day-08-haskell/test/Spec.hs b/day-08-haskell/test/Spec.hs new file mode 100644 index 0000000..cd4753f --- /dev/null +++ b/day-08-haskell/test/Spec.hs @@ -0,0 +1,2 @@ +main :: IO () +main = putStrLn "Test suite not yet implemented"