99 questions/Solutions/87

From HaskellWiki
< 99 questions‎ | Solutions
Revision as of 23:25, 5 March 2011 by Jagbolanos (talk | contribs) (Depth First Search in a graph)
(diff) ← Older revision | Latest revision (diff) | Newer revision → (diff)
Jump to navigation Jump to search
The printable version is no longer supported and may have rendering errors. Please update your browser bookmarks and please use the default browser print function instead.
type Node = Int
type Edge = (Node,Node)
type Graph = ([Node],[Edge])
depthfirst :: Graph -> Node -> [Node]
depthfirst (v,e) n
   | [x|x<-v,x==n] == [] = []
   | otherwise = dfrecursive (v,e) [n]
dfrecursive :: Graph -> [Node] -> [Node]
dfrecursive ([],_) _ = []
dfrecursive (_,_) [] = []
dfrecursive (v,e) (top:stack)
   | [x|x<-v,x==top] == [] = dfrecursive (newv, e) stack
   | otherwise = top : dfrecursive (newv, e) (adjacent ++ stack)
   where
       adjacent = [x | (x,y)<-e,y==top] ++ [x | (y,x)<-e,y==top]
       newv = [x|x<-v,x/=top]


Call it:

depthfirst ([1,2,3,4,5],[(1,2),(2,3),(1,4),(3,4),(5,2),(5,4)]) 1