adventofcode
/
2017
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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.IntMap (IntMap, findWithDefault)
import Data.IntMap (IntMap, findWithDefault, keys)
import Data.IntSet (IntSet, member, notMember, insert, delete, empty, size, findMin)
import qualified Data.IntMap as M (fromList)
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 = do
keyvals <- fmap (map parseLine . lines) $ readFile "12.txt"
let hashkeys = S.fromList . fst . unzip $ keyvals
hashmap = M.fromList keyvals
hashmap <- fmap (M.fromList . map parseLine . lines) $ readFile "12.txt"
print $ size $ visit hashmap 0 empty
print $ countGroups hashmap 0 hashkeys
print $ countGroups hashmap 0 . S.fromList . keys $ hashmap