Question

我试图在Hackerrank中解决这个graph problem，这是我到目前为止所做的。我使用Python字典来表示图形，并让我的DFS函数返回它遍历的连接组件的长度。我的代码通过了第一个测试用例，但是给了我一些其他测试用例的运行时错误。这是一个优化问题吗？如果是这样，我应该尝试优化我的代码的哪一部分？或者我应该尝试不同的方法吗？

import sys
n = input()
# Graph
g = {k:set() for k in xrange(2*n)}

# DFS stuff
visited = set()

def dfs(g,s,S=set(),dfs_sum=0):
    S.add(s)
    dfs_sum +=1
    for i in g[s]:
        if i in S: continue
        dfs_sum = dfs(g,i,S,dfs_sum) 
    return dfs_sum

# Building the graph
for i in xrange(n):
    a,b = map(int,raw_input().split())
    g[a].add(b)
    g[b].add(a)

# Getting the max and min lengths of the connected components
max_len, min_len = 0, sys.maxint
for i in xrange(n):
    if i not in visited:
        v = dfs(g,i,visited)
        if v > 1: # Ignore single nodes
            max_len, min_len = max(max_len, v), min(min_len, v)
print("{} {}".format(min_len,max_len))

Answer 1

由于可能有2 * 15000个节点，因此您可能exceeding maximum recursion depth。您可以使用非递归方法（如BFS，迭代DFS或disjoint-set data structure）来克服此问题。

另一种方法是增加递归限制：

sys.setrecursionlimit(30000)

此外，问题是使用基于1的索引，因此您应该更改此行：

g = {k:set() for k in xrange(2*n)}

到

g = {k:set() for k in xrange(2*n+1)}

尝试在图表中查找组件时出现运行时错误

1 个答案: