Haskell - 生成节点之间的所有路径

时间:2012-06-23 09:07:41

标签: haskell

我需要构建一个函数,它返回某些节点之间的所有路径。

connect :: Int -> Int-> [[(Int,Int)]]

Data.Graph库为我提供了有用的函数'buildG',它为我构建了图形。如果我打电话

let g = buildG (1,5) [(1,2),(2,3),(3,4),(4,5),(2,5)]

我将获得一个数组,其中每个节点都映射到他的邻居。 一个例子:

g!1 = [2]
g!2 = [3,5] 
..
g!5 = []

我试图使用列表推导来做到这一点,但我对haskell并不是很好,我也是 得到打字错误,我无法修复。

connect x y g 
    | x == y = []
    | otherwise = [(x,z) | z <- (g!x), connect z y g] 

此刻我不需要担心周期。这是我想要的:

connect 1 5 g = [[(1,2),(2,3),(3,4),(4,5)],[(1,2),(2,5)]]

1 个答案:

答案 0 :(得分:4)

递归思考。从se的路径由第一条边(s,t)和从te的路径组成,除非s == e,其中路径应为空。所以第一次尝试是

connect x y g
    | x == y    = [[]]
    | otherwise = [(x,t):path | t <- g!x, path <- connect t y g]

从节点到自身的所有符合条件的路径列表是具有单个元素[]的列表,在其他情况下,我们通过上面的逻辑获取所有符合条件的路径的列表,选择第一个边缘和从最后找到路径。

问题是循环会导致它挂起。为避免这种情况,您必须记住已访问过的节点,而不是从中探索路径:

connect x y g = helper x y g [x]
  where
    helper a b g visited
        | a == b    = [[]]
        | otherwise = [(a,c):path | c <- g!a, c `notElem` visited, path <- helper c b g (c:visited)]