python 广度优先搜索 遍历图中的点

Question

mapmodel=[
[0,1,1,-1,1],
[1,0,-1,1,-1],
[1,-1,0,-1,1],
[-1,1,-1,0,-1],
[1,-1,1,-1,0]
]

flag=[1,0,0,0,0]

def dfs(current,sumpoint):

if sumpoint==5:
print sumpoint,flag
for i in range(5):
if mapmodel[current][i]==1 and flag[i]==0:
sumpoint=sumpoint+1
flag[i]=i
dfs(i,sumpoint)
return

dfs(0,1)

我想要遍历图里面的点，但总是得不到正确结果，不知道哪里出问题了，请大家指教

Answer 1

试下这个，flag和module和你上面的一样，然后dfs（0）

 def dfs(current):
    for i in range(5):
        if mapmodel[current][i]==1 and flag[i]==0:
            flag[i]=1
            print i,flag
            dfs(i)