定一个二叉树的根节点 root
,返回 它的 中序 遍历 。
在二叉树的遍历种,我们可以使用栈迭代,使用栈的出栈和入栈方法实现中序遍历,为了方便判断,我们为每个节点赋予颜色。核心思想是:
⚪使用颜色标记节点的状态,新节点为白色,已访问的节点为灰色。
⚪如果遇到的节点为白色,则将其标记为灰色,然后将其右子节点、自身、左子节点依次入栈。
⚪如果遇到的节点为灰色,则将节点的值输出。
具体Python代码实现如下:
def inorderTraversal(root):
WHITE,GRAY=0,1
res=[]
stack=[(WHITE,root)]
while stack:
color,node=stack.pop()
#先判断节点是否为空
if node is None:continue
#再判断该节点是什么颜色
if color==WHITE:
#先序遍历,入栈顺序与遍历顺序相反
stack.append((WHITE,node.right))
stack.append((GRAY,node))
stack.append((WHITE,node.left))
else:
#如果是已经访问了则添加值
res.append(node.val)
return res
但是我们发现虽然为每个节点添加颜色会使得遍历判断起来很便捷,但是有没有一种更便捷的方法呢,当然有。当将子节点按照中序入栈时我们只需要将子节点的根节点设置为int值子节点的儿子设置为Treenode类型进行判断即可。代码如下:
def inorderTraversal(root):
stack,res=[root],[]
while stack:
i=stack.pop
#直接判断节点是什么类型
if isinstance(i,TreeNode):
stack.extend([i.right,i.val,i.left])
elif isinstance(i,int):
res.append(i)
return res
我们观察到两次方法一次用了extend一次用了append,下面介绍append与extend方法的区别:
1.append可以添加单个元素,也可以添加可迭代对象,但是extend只能添加可迭代对象:
arr = [1,2,3,4]
arr.append(5)
print(arr)
Out:[1, 2, 3, 4, 5]
arr.extend(5)
Out:TypeError: 'int' object is not iterable
2.在添加可迭代对象,append在添加后不改变添加项的类型,添加之前是什么类型,添加之后就是什么类型;而extend在添加后,会将添加项进行迭代,迭代的元素挨个添加到被添加的数组中:
arr = [1,2,3,4]
item = [5,6,7]
arr.append(item)
print(arr)
Out:[1, 2, 3, 4, [5, 6, 7]] #不会改变添加项的类型
arr.extend(item)
print(arr)
Out:[1, 2, 3, 4, 5, 6, 7]# 添加项进行迭代