ionchy
/
adventofcode
1
0
Fork 0
Code

Day 12 - use array constructor instead of graphFromEdges, which needs and produces some extra stuff.

This commit is contained in:
Jonathan Chan 2017-12-12 15:30:16 -08:00
parent dd002c175c
commit 492ccb0a5d
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
12_graph.hs
View File

@ -1,13 +1,14 @@
import Data.List.Split (splitOn) import Data.List.Split (splitOn)
import Data.Graph (graphFromEdges, reachable, scc) import Data.Graph (reachable, scc)
import Data.Array (array)
parseLine :: String -> (Int, Int, [Int]) parseLine :: String -> (Int, [Int])
parseLine str = parseLine str =
let src : dests : [] = splitOn " <-> " str let src : dests : [] = splitOn " <-> " str
in (read src, read src, map read $ splitOn ", " dests) in (read src, map read $ splitOn ", " dests)
main :: IO () main :: IO ()
main = do main = do
(graph, _, _) <- fmap (graphFromEdges . map parseLine . lines) $ readFile "12.txt" graph <- fmap (array (0, 1999) . map parseLine . lines) $ readFile "12.txt"
print $ length $ reachable graph 0 print $ length $ reachable graph 0
print $ length $ scc graph print $ length $ scc graph