python - 如何使用 DFS 正确检测循环？

Question

这是我在图中检测循环的代码：

seen = set()

def check(table,node):

    seen.add(node)

    #traverse all neighors of the current node

    for neigh in table[node]:
        print(neigh, seen)

        if neigh not in seen:
            check(table,neigh)
        else:
            return False
    return True



table1 = {'A':['B'],'B':['A']}

print(check(table1,'A'))

为什么它总是返回 True？即使存在循环，else 子句也不会执行。我错过了什么吗？谢谢！

标签： pythondepth-first-search

Answer 1

我认为问题在于您需要返回递归调用check（在 if 语句之后）使您的代码看起来像：

seen = set()

def check(table,node):

    seen.add(node)

    #traverse all neighors of the current node

    for neigh in table[node]:
        print(neigh, seen)

        if neigh not in seen:
            return check(table,neigh)
        else:
            return False
    return True



table1 = {'A':['B'],'B':['A']}

print(check(table1,'A'))

您的原始代码会发生什么情况是您使用节点“B”递归调用 check，它返回 false，但您不对该返回值执行任何操作，因此在顶层它退出 for 循环并向用户返回 true。希望这可以帮助你！