ionchy
/
adventofcode
1
0
Fork 0
Code

Day 12 - minor refactoring

This commit is contained in:
Jonathan Chan 2017-12-11 23:05:16 -08:00
parent a90a59c846
commit 46ca9a3347
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
12.hs
View File

@ -1,5 +1,5 @@
import Data.List.Split (splitOn) import Data.List.Split (splitOn)
import Data.IntMap (IntMap, findWithDefault) import Data.IntMap (IntMap, findWithDefault, keys)
import Data.IntSet (IntSet, member, notMember, insert, delete, empty, size, findMin) import Data.IntSet (IntSet, member, notMember, insert, delete, empty, size, findMin)
import qualified Data.IntMap as M (fromList) import qualified Data.IntMap as M (fromList)
import qualified Data.IntSet as S (fromList, null) import qualified Data.IntSet as S (fromList, null)
@ -25,8 +25,6 @@ countGroups hashmap count hashset = if S.null hashset then count else
main :: IO () main :: IO ()
main = do main = do
keyvals <- fmap (map parseLine . lines) $ readFile "12.txt" hashmap <- fmap (M.fromList . map parseLine . lines) $ readFile "12.txt"
let hashkeys = S.fromList . fst . unzip $ keyvals
hashmap = M.fromList keyvals
print $ size $ visit hashmap 0 empty print $ size $ visit hashmap 0 empty
print $ countGroups hashmap 0 hashkeys print $ countGroups hashmap 0 . S.fromList . keys $ hashmap