Day 12: Forgot to use the formatter
This commit is contained in:
parent
5e171b20c0
commit
12d28015d4
|
@ -1,5 +1,5 @@
|
|||
import Data.List.Split
|
||||
import Data.List
|
||||
import Data.List.Split
|
||||
import Linear.V3
|
||||
|
||||
main :: IO ()
|
||||
|
@ -18,32 +18,41 @@ toV3 = map (\[x,y,z]->V3 x y z)
|
|||
|
||||
parseContent :: String -> [[Int]]
|
||||
parseContent = map (map read . splitOn ",") . lines . removeJunk
|
||||
where removeJunk xs = [ x | x <- xs, x `notElem` " <>xyz=" ]
|
||||
where
|
||||
removeJunk xs = [x | x <- xs, x `notElem` " <>xyz="]
|
||||
|
||||
step :: (Num a, Eq a) => ([a], [a]) -> ([a], [a])
|
||||
step (moons, vel) = (zipWith (+) moons newVel, newVel)
|
||||
where dVel = gravity moons
|
||||
where
|
||||
dVel = gravity moons
|
||||
newVel = zipWith (+) vel dVel
|
||||
gravity xs = [sum [signum $ x - y | x <- xs, y /= x] | y <- xs]
|
||||
|
||||
energy :: ([V3 Int], [V3 Int]) -> Int
|
||||
energy (x,y) = sum $ zipWith (*) (geten x) (geten y) where
|
||||
energy (x, y) = sum $ zipWith (*) (geten x) (geten y)
|
||||
where
|
||||
geten = map (sum . abs)
|
||||
|
||||
findPeriod :: (Num a, Eq a) => ([a], [a]) -> ([a], [a]) -> Int -> Int
|
||||
findPeriod x a n = if x'==a then n else findPeriod x' a (n+1)
|
||||
where x' = step x
|
||||
findPeriod x a n =
|
||||
if x' == a
|
||||
then n
|
||||
else findPeriod x' a (n + 1)
|
||||
where
|
||||
x' = step x
|
||||
|
||||
part2 :: (Num a, Eq a) => ([[a]], [[a]]) -> Int
|
||||
part2 (moons, vel) = lcm' periods
|
||||
where m = transpose moons
|
||||
where
|
||||
m = transpose moons
|
||||
v = transpose vel
|
||||
periods = zipWith (curry findPeriod') m v
|
||||
findPeriod' x = findPeriod x x 1
|
||||
|
||||
lcm' :: [Int] -> Int
|
||||
lcm' xs = product $ zipWith (^) nums maxElems
|
||||
where nums = nub $ concat decomp
|
||||
where
|
||||
nums = nub $ concat decomp
|
||||
decomp = map decomposition xs
|
||||
maxElems = [maximum $ map (length . elemIndices x) decomp | x <- nums]
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user