树似乎在Python中错误地构建

时间:2016-05-10 18:29:57

标签: python python-3.x data-structures tree nodes

data = {'murtaza':('arnav', 'ohjun'),
        'ohjun':('arnav', 'murtaza'),
        'arnav':('ohjun', 'murtaza')}

class node:
    predecessors=[]
    nexts=[]
    student=''
    def __init__(self, student='', predecessors=[]):
        self.student=student
        self.predecessors=predecessors
    def grow(self, max=6, depth=0):
        if not self.student in self.predecessors:
            self.predecessors.append(self.student)
            for pref in data[self.student]:
                next=node(pref, self.predecessors)
                print(depth, self.predecessors, self.student, pref)
                next.grow(max, depth=depth+1)
                self.nexts.append(next)
        else:
            return

所以,这是我的节点的数据和类定义。当我在节点对象上调用grow()方法时,我期望发生的事情如下:它在数据中查找学生姓名,然后查找与该学生相关的每个姓名(即他们的偏好,因此{ {1}})创建一个新节点,将其添加到从当前节点分支出来的节点,然后在该新节点上调用for pref in data[self.student]。然后该方法将继续构建树,除非学生在序列中出现两次,因此检查grow()

问题似乎是前辈们在树中没有被正确记住。出于某种原因,兄弟节点突然得到了他们孩子的所有前辈。这是程序的输出:

if not self.student in self.predecessors

我希望如下:

0 ['murtaza'] murtaza arnav
1 ['murtaza', 'arnav'] arnav ohjun
2 ['murtaza', 'arnav', 'ohjun'] ohjun arnav
2 ['murtaza', 'arnav', 'ohjun'] ohjun murtaza #as expected up to here
1 ['murtaza', 'arnav', 'ohjun'] arnav murtaza
0 ['murtaza', 'arnav', 'ohjun'] murtaza ohjun

我是否理解了错误的代码,或者我对算法的理解是错误的?或者我的实施错了?我真的以为我理解如何制作树,但这似乎并不像我认为的那样工作。

编辑:我应该补充说主体看起来像这样:

0 ['murtaza'] murtaza arnav
1 ['murtaza', 'arnav'] arnav ohjun
2 ['murtaza', 'arnav', 'ohjun'] ohjun arnav
2 ['murtaza', 'arnav', 'ohjun'] ohjun murtaza 
1 ['murtaza', 'arnav'] arnav murtaza
0 ['murtaza'] murtaza ohjun
1 ['murtaza', 'ohjun'] ohjun arnav
2 ['murtaza', 'ohjun', 'arnav'] arnav ohjun
2 ['murtaza', 'ohjun', 'arnav'] arnav murtaza
1 ['murtaza', 'ohjun'] ohjun murtaza

1 个答案:

答案 0 :(得分:2)

这一行:

next=node(pref, self.predecessors)

不会复制self.predecessors,而是传递对指定的list对象的引用。同样,这一行:

    self.predecessors=predecessors

不会复制predecessors。因此,所有node个对象都在同一个列表中运行;当一个对象调用.append()时,所有对象都会被调用。 predecessor列表已更新。

解决方案是用copy.deepcopy()替换其中一个调用,如下所示:

    self.predecessors = copy.deepcopy(predecessors)

根据您的精确数据结构,您可能需要copy.copy()

此外,可能与您的问题无关,您在predecessors中为__init__()提供的默认值将无法按预期运行。 (见here)。试试这个:

def __init__(self, student='', predecessors=None):
    self.student=student
    if predecessors is None:
        self.predecessors = []
    else:
        self.predecessors=copy.deepcopy(predecessors)