递归TreeItems搜索

时间:2014-08-09 14:37:06

标签: python

我有QTreeWidget嵌套的父子项:

Tree:
    parent1:
           childA1
           childA2
           childA3:
                  childB1
                  childB2:
                         childC1

使用:

    allItems=[]
    for i in range(Tree.topLevelItemCount()):
        item=Tree.topLevelItem(i)
        allItems.add(item)

我访问了第一级项目" A":childA1, childA2 and childA3

现在我迭代第一级" A"访问孩子的项目 - 第二级项目" B":

    for i in range(Tree.topLevelItemCount()):
        item=Tree.topLevelItem(i)

        for m in range(item.childCount()):
            childItem=item.child(m)
            allItems.add(childItem)

在第二级,我不知道是否有任何第三级项目" C"下面。 如何确保函数的下降与嵌套项目一样深,将项目添加到allItems列表直到它结束?

2 个答案:

答案 0 :(得分:0)

您可能最好使用递归实现,尤其是因为在您的情况下,递归深度的要求似乎非常低。

如果你的树的高度超过1000,请注意Python有一个默认的递归限制[可以通过sys.setrecursionlimit(limit)设置限制来覆盖,尽管从CPython(python')开始这样做是不鼓励的。 s默认实现)不优化深度递归,因此不优化默认限制]

这种递归实现有很多指南。 This此处的问题可以针对您的特定应用进行修改。

如果你想要一个非递归的实现,下面可能适用于你的情况[注意我还没有测试过相同的]。这只是为了让你得到一个想法。

allItems=[]
buff = []

# Populating the buffer with all the top level items
for i in range(Tree.topLevelItemCount()):
    item=Tree.topLevelItem(i)
    buff.append(item)

while len(buff) > 0:
    # Take the first item out of the buffer by popping it
    current_item = buff.pop(buff[0])

    # Put all the childs of current_item in the buffer
    for child_no in range(current_item.childCount):
        buff.append(current_item.child(child_no))

    # Append the current_item to the allItems list
    allItems.append(current_item)

答案 1 :(得分:0)

下面的代码使用嵌套项创建QTreeWidget

树的getItemsRecursively()方法扫描树,直到它到达底部项目,返回树中所有项目的列表。

enter image description here

from PyQt4 import QtCore, QtGui
app = QtGui.QApplication([])

class Tree(QtGui.QTreeWidget):
    def __init__(self, *args, **kwargs):
        super(Tree, self).__init__()
        for name in ['Item_A0', 'Item_A1', 'Item_A2']: 
            itemA=QtGui.QTreeWidgetItem([name])

            self.addTopLevelItem(itemA)
            if name=='Item_A1':
                item_B0=QtGui.QTreeWidgetItem(['Item_B0'])
                itemA.insertChild(0, item_B0)
                itemA.insertChild(1, QtGui.QTreeWidgetItem(['Item_B1']))

                item_C0=QtGui.QTreeWidgetItem(['Item_C0'])
                item_B0.insertChild(0, item_C0)
                item_B0.insertChild(0, QtGui.QTreeWidgetItem(['Item_C1']))

        self.resize(360,240)
        self.show()

    def getItemsRecursively(self):
        total=[]
        def serchChildItem(item=None):
            if not item: return
            for m in range(item.childCount()): 
                childItem=item.child(m)                
                if not childItem: continue
                total.append(childItem)   
                serchChildItem(childItem)

        for i in range(self.topLevelItemCount()):
            item=self.topLevelItem(i)
            if not item: continue
            total.append(item)
            serchChildItem(item)
        return total

tree=Tree()
total=tree.getItemsRecursively()
print 'Total number of Tree items: %s'%len(total)
sys.exit(app.exec_())