Day 8: Resonant Collinearity

Megathread guidelines

  • Keep top level comments as only solutions, if you want to say something other than a solution put it in a new post. (replies to comments can be whatever)
  • You can send code in code blocks by using three backticks, the code, and then three backticks or use something such as https://topaz.github.io/paste/ if you prefer sending it through a URL

FAQ

  • VegOwOtenks@lemmy.world
    link
    fedilink
    English
    arrow-up
    5
    ·
    edit-2
    20 days ago

    Haskell

    I overslept 26 minutes (AoC starts at 06:00 here) which upsets me more than it should.
    I thought this one was going to be hard on performance or memory but it was surprisingly easy.

    import Control.Arrow hiding (first, second)
    import Data.Bifunctor
    
    import Data.Array.Unboxed (UArray)
    
    import qualified Data.List as List
    import qualified Data.Set as Set
    import qualified Data.Array.Unboxed as Array
    
    parse :: String -> UArray (Int, Int) Char
    parse s = Array.listArray ((1, 1), (n, m)) . filter (/= '\n') $ s :: UArray (Int, Int) Char
    
            where
                    n = takeWhile   (/= '\n') >>> length $ s
                    m = List.filter (== '\n') >>> length >>> pred $ s
    
    groupSnd:: Eq b => (a, b) -> (a', b) -> Bool
    groupSnd = curry (uncurry (==) <<< snd *** snd)
    
    cartesianProduct xs ys = [(x, y) | x <- xs, y <- ys]
    
    calculateAntitone ((y1, x1), (y2, x2)) = (y1 + dy, x1 + dx)
            where
                    dy = y1 - y2
                    dx = x1 - x2
    
    antennaCombinations = Array.assocs
            >>> List.filter (snd >>> (/= '.'))
            >>> List.sortOn snd
            >>> List.groupBy groupSnd
            >>> map (map fst)
            >>> map (\ xs -> cartesianProduct xs xs)
            >>> map (filter (uncurry (/=)))
    
    part1 a = antennaCombinations
            >>> List.concatMap (map calculateAntitone)
            >>> List.filter (Array.inRange (Array.bounds a))
            >>> Set.fromList
            >>> Set.size
            $ a
    
    calculateAntitones ((y1, x1), (y2, x2)) = iterate (bimap (+dy) (+dx)) (y1, x1)
            where
                    dy = y1 - y2
                    dx = x1 - x2
    
    part2 a = antennaCombinations
            >>> List.map (map calculateAntitones)
            >>> List.concatMap (List.concatMap (takeWhile (Array.inRange (Array.bounds a))))
            >>> Set.fromList
            >>> Set.size
            $ a
    
    main = getContents
            >>= print
            . (part1 &&& part2)
            . parse
    
    • lwhjp@lemmy.sdf.org
      link
      fedilink
      arrow-up
      2
      ·
      20 days ago

      D’oh. Computing antinodes in a single direction and permuting pairs is a much neater approach that what I did!