diff options
author | Determinant <ted.sybil@gmail.com> | 2017-05-31 21:23:39 -0400 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2017-05-31 21:23:39 -0400 |
commit | 741e26dabbccb5821c79736590ad03286fe04f84 (patch) | |
tree | 73533fe49b5ee59d75728ba0a0fc2713e07c8f4d /80-89/88.hs | |
parent | 1a19a80d8c756559e108a46129fac9489ad43d28 (diff) |
one prob left for v9
Diffstat (limited to '80-89/88.hs')
-rw-r--r-- | 80-89/88.hs | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/80-89/88.hs b/80-89/88.hs new file mode 100644 index 0000000..6c1123c --- /dev/null +++ b/80-89/88.hs @@ -0,0 +1,22 @@ +import Data.List (partition, foldl') + +data Graph a = Graph [a] [(a, a)] deriving (Show, Eq) + +connComp :: Eq a => Graph a -> [[a]] + +connComp (Graph v e) = fst $ foldl' + (\(comps, vis) u -> if u `elem` vis then (comps, vis) + else let (s, vis') = dfs u vis in + (s:comps, vis')) + ([], []) v + where adjust u (a, b) + | a == u = [b] + | b == u = [a] + | otherwise = [] + dfs u vis + | u `elem` vis = ([], vis) + | otherwise = let vs = (e >>= (adjust u)) in + foldl' (\(s, vis') v -> + let (s', vis'') = dfs v vis' in + (s ++ s', vis'')) + ([u], u:vis) vs |