在 python 中优化 DFS

Question

我正在学习算法并且正在做这个problem，它正在寻找区域的数量。我尝试了使用 python 的深度优先搜索方法，但我收到了超出调用堆栈的错误。谁能告诉我实施中的缺陷是什么以及如何克服它？代码是：

def findCircleNum(A):

    count = 0

    def dfs(row, column):
        if row < 0 or column >= len(A[0]) or row >= len(A) or column < 0:
            return

        if A[row][column] == 0:
            return

        A[row][column] = 0

        for i in range(row-1, row+2):
            if i >= 0 and i<len(A) and A[i][column] == 1:
                dfs(i, column)
        for i in range(column-1, column+2):
            if i >= 0 and i<len(A[0]) and A[row][i] == 1:
                dfs(row, i)




    for row in range(len(A)):
        for column in range(len(A[0])):
            if A[row][column] == 1:
                dfs(row, column)
                count += 1


    return count


findCircleNum(m)

失败的输入是一个全 1 的 100x100 矩阵

错误是get是：

RuntimeError: maximum recursion depth exceeded

谢谢！

Answer 1

为什么不做一个标准的 DFS，同时用一组跟踪访问的顶点（学生）？问题陈述说矩阵的最大大小为 200x200，因此假设它为 1000，您不必担心递归限制。使用集合进行跟踪也会使代码更简单：

def findCircleNum(A):
    count = 0
    visited = set()

    def dfs(student):
        if student in visited:
            return

        visited.add(student)
        for i, friend in enumerate(A[student]):
            if friend:
                dfs(i)

    for student in range(len(A)):
        # Note we want to track circles, student without any friend won't count
        if student not in visited and any(A[student]):
            dfs(student)
            count += 1

    return count

编辑 有问题的代码看起来像是在进行 DFS 时将边视为顶点。这也可以解释递归深度的问题，因为 100 个带循环但没有多重边的顶点的无向图具有最大 (100 * 101) / 2 = 5050 条边。