diff --git a/app/2.hs b/app/2.hs index 4769fe3..fa88c1a 100644 --- a/app/2.hs +++ b/app/2.hs @@ -15,15 +15,10 @@ solve1 = solve2 :: [[Int]] -> Int solve2 = let r [] = [[]] - r (r':rs) = rs:((r':) <$> r rs) - di a = all (\a' -> a' < 4 && not (a' <= 0)) a || - all (\a' -> a' > -4 && not (a' >= 0)) a in - sum . ((\a -> - if any (\b -> di $ zipWith (flip (-)) b (tail b)) a - then 1 else 0) . + r (r':rs) = rs:((r':) <$> r rs) in + sum . (((\a -> if a > 0 then 1 else 0) . solve1 ) . r <$>) - main :: IO () main = readFile "inputs/2" <&> parse >>= \i -> print (solve1 i) >>