Python：复杂的迭代

Question

我已经看到这段代码迭代了一个类的某些成员（如果它们存在的话）。值得注意的是，在二叉树中，遍历子节点直到没有更多子节点。

二叉树定义为..

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

他们已经像这样迭代它：

# type root : TreeNode
def iterateTree(self, root):
    level_list = [root]

    while level_list:

        for item in level_list:
             print(item.val)

        # This iterable seems really complicated for me to understand how they came up with this
        level_list = [child for node in level_list for child in (node.left, node.right) if child]

我不确定他们是如何想出那条线来遍历左右节点的，我永远不会当场想出那条线……我该如何剖析这条线？

Answer 1

如果我没记错的话，这个语句是一种创建列表的pythonic和简写方式。

 # This iterable seems really complicated for me to understand how they came up with this
   level_list = [child for node in level_list for child in (node.left, node.right) if child]

这基本上是执行以下一组行的简写方式：

for node in level_list:
    for child in (node.left, node.right):
        if child:
            level_list.append(child)

理解这个速记语句的诀窍是查看外围边界符号，在这种情况下它们是[ 和]。它与python中的列表序列标识。由于列表中有迭代器（for 循环），我们基本上是在所述列表中创建或添加元素（变量child）。

/ogs

Answer 2

如下：

for node in level_list:
    for child in (node.left, node.right):
        if child:
            child