python - 嵌套列表的mptt

时间:2011-07-12 12:32:48

标签: python recursion mptt

我已经阅读了一些关于mptt解决方案的内容,我正在尝试实现某种将嵌套列表转换为mptt列表的方法。我知道,在这种格式中它没用,但嵌套列表可以通过类子类或其他任何东西轻松更改。

我的代码是:

tree = [[1,[[11,[111, 112, 113]],[12,]]],[2,[[21,[[211,[2111,]],]],]]]

class node():
    def __init__(self, value, lft):
        self.value = value
        self.lft = lft

    def add_right(self, rgt):
        self.rgt = rgt

    def __str__(self):
        return "%s | %s | %s" %(self.value, self.lft, self.rgt)

def numerate(table, tree, counter):
    for item in tree:
        if type(item) == type(1):
            table.append(node(item, counter))
            index = len(table)
            counter += 1
        else:
            table.append(node(item[0], counter))
            index = len(table)
            counter += 1

            if len(item) > 1:
                (table, counter) = numerate(table, item[1], counter)

        table[index-1].add_right(counter)
    return (table, counter)

table = numerate([], tree, 0)[0]

for item in table:
    print item

但结果是:

1 | 0 | 6
11 | 1 | 5
111 | 2 | 3
112 | 3 | 4
113 | 4 | 5
12 | 5 | 6
2 | 6 | 10
21 | 7 | 10
211 | 8 | 10
2111 | 9 | 10

我看到在递归中有正确值的东西不正常。我认为python中的这种格式有时可能有用,所以为它编写一些代码会很好。

1 个答案:

答案 0 :(得分:0)

通过添加全局迭代器解决了这个问题。

如果有人感兴趣,那就是新的计算功能,实际上工作正常。

global_iterator = 0

def numerate(table, tree):
    global global_iterator
    for item in tree:
        global_iterator += 1


        if type(item) == type(1):
            table.append(node(item, global_iterator))
            index = len(table)
        else:
            table.append(node(item[0], global_iterator))
            index = len(table)

            if len(item) > 1:
                table.append(numerate(table, item[1]))

        global_iterator += 1

        table[index-1].add_right(global_iterator)

    return table

table = numerate([], tree)

node()不会改变。