Tidying

diff --git a/advent01/Main.hs b/advent01/Main.hs
index 5cf7d6a9adb4c7b65c3e980f49effa477583639c..5a45a1e272caea1a53c2d90a3968e0978cdb69c9 100644 (file)
--- a/advent01/Main.hs
+++ b/advent01/Main.hs
@@ -3,11 +3,12 @@
 import System.Environment
 import Data.List
 import Data.List.Split
+import Data.Ord
 
 main :: IO ()
 main = 
   do  dataFileName <- getDataFileName
-      numStrs <- readFile dataFileName -- "data/advent01.txt"
+      numStrs <- readFile dataFileName
       let calories = fmap (fmap (read @Int)) $ splitWhen null $ lines numStrs
       print $ part1 calories
       print $ part2 calories
@@ -26,5 +27,6 @@ part1 :: [[Int]] -> Int
 part1 = maximum . fmap sum
 
 part2 :: [[Int]] -> Int
-part2 = sum . take 3 . reverse . sort . fmap sum
+-- part2 = sum . take 3 . reverse . sort . fmap sum
+part2 = sum . take 3 . sortOn Down . fmap sum