用于生成图的递归算法

时间:2011-02-25 12:51:26

标签: python graph

我有一些数据库对象作为依赖项完全相互链接。我想要做的是编写一个算法来检索该信息,并将所有这些表示为图形。现在一个伪代码应该为我做的技巧,然后我应该能够编写python实现。这看起来像一个递归算法,这就是我被困住的地方!

 Input (Obj)
 list = obj.getDependencies():
 if list is empty:
   return obj
 else
   for items in list:
     list1 = item.getDependencies()
     if list1 is empty:
       return item
     else:
       list2 = item.getDependencies()
       ......

我的思绪在这一点上爆炸!我怎样才能重写这个算法

1 个答案:

答案 0 :(得分:1)

如果我理解正确,您只需要树的叶节点(那些没有更多依赖关系的节点)。是这样的吗?使用辅助结构使其可运行的示例:

class Struct:
    def __init__(self, **entries): 
        self.__dict__.update(entries)        

obj = Struct(
    name="1",
    get_dependencies=lambda: [
        Struct(name="11", get_dependencies=lambda: []), 
        Struct(name="12", get_dependencies=lambda: [
            Struct(name="121", get_dependencies=lambda: [])
        ])
    ])

def get_all_dependencies(obj):
    ds = obj.get_dependencies()
    if not ds:
        yield obj
    for o in ds:
        for o2 in get_all_dependencies(o):
            yield o2

print [x.name for x in get_all_dependencies(obj)]
# ['11', '121']

如果你喜欢itertools可以实现的紧凑代码,那么具有完全相同想法的不同实现:

import itertools

def flatten(it):
    return itertools.chain.from_iterable(it)

def get_all_dependencies(obj):
    ds = obj.get_dependencies()
    return ([obj] if not ds else flatten(get_all_dependencies(o) for o in ds))

print [x.name for x in get_all_dependencies(obj)]
# ['11', '121']