使用python实现拓扑排序的错误

时间:2015-02-25 09:49:31

标签: python topological-sort

更新

我没有在下面的代码中使用嵌套函数,但答案是错误的: {'s': 4, 't': 4, 'w': 4, 'v': 4}

我认为问题出现在递归调用中,n在函数回调时不会减少,但我不知道如何修改它。

import unittest
#dfs
def dfs(graph,start,visited,n,topo_order):
        visited.add(start)
        for w in graph[start]:
            if w not in visited:
                dfs(graph,w,visited,n,topo_order)
        topo_order[start]=n
        n-=1
def topological_sort(graph):
    visited=set()
    n=len(graph)
    topo_order={}
    for v in graph:
        if v not in visited:
            dfs(graph,v,visited,n,topo_order)
    print(topo_order)
class topological_sort_test(unittest.TestCase):
    def test(self):
        file=[]
        with open('topological_test.txt') as f:
            data=f.read()
        data=data.split('\n')
        data=[i.split() for i in data]        
        G={}
        for lst in data:
            G[lst[0]]=lst[1:]
        topological_sort(G)
if __name__=='__main__':
    unittest.main()    

1 个答案:

答案 0 :(得分:1)

def dfs(graph,start)

函数声明为变量创建局部范围。要从外部访问任何变量,您需要将其作为参数传递给此函数。