2017-12-12 06:43:01 +00:00
|
|
|
import Data.List.Split (splitOn)
|
2017-12-16 19:12:40 +00:00
|
|
|
import Data.Graph (reachable, scc)
|
|
|
|
import Data.Array (array)
|
2017-12-12 06:43:01 +00:00
|
|
|
|
|
|
|
parseLine :: String -> (Int, [Int])
|
|
|
|
parseLine str =
|
|
|
|
let src : dests : [] = splitOn " <-> " str
|
|
|
|
in (read src, map read $ splitOn ", " dests)
|
|
|
|
|
|
|
|
main :: IO ()
|
|
|
|
main = do
|
2017-12-16 19:12:40 +00:00
|
|
|
graph <- fmap (array (0, 1999) . map parseLine . lines) $ readFile "12.txt"
|
|
|
|
print $ length $ reachable graph 0
|
|
|
|
print $ length $ scc graph
|