94.二叉树的中序遍历，extend()与append()的区别，力扣/Leetcode算法题，面试算法题，Python3，Python，二叉树,

最新推荐文章于 2024-08-14 11:57:07 发布

温涛

最新推荐文章于 2024-08-14 11:57:07 发布

阅读量292

点赞数

分类专栏：学习文章标签：算法面试 leetcode python 数据结构

本文链接：https://blog.csdn.net/u013373155/article/details/126675601

版权

学习专栏收录该内容

16 篇文章 0 订阅

订阅专栏

本文探讨了两种优化方法实现二叉树的中序遍历：一是利用颜色标记节点状态，二是通过智能判断简化节点类型判断。两种方法对比并介绍了Python代码实现，同时解析了append和extend在操作列表时的不同之处。

摘要由CSDN通过智能技术生成

定一个二叉树的根节点 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]# 添加项进行迭代