二叉树的先序(preorder),中序(inorder),后序(postorder)的遍历(python)

本文深入探讨了二叉树的三种遍历方式:先序、中序和后序遍历,介绍了每种遍历的应用场景及LeetCode相关算法题的解法。通过递归和迭代方法,以及一种高效的颜色标记法,使读者能更直观地理解和实现遍历算法。
部署运行你感兴趣的模型镜像

本篇文章主要是总结二叉树的三种遍历方式,和相关leetcode算法题的解法。
跟线性的数据结构(矩阵,列表,队列,栈···)不同,树的遍历可以有不同的遍历方式。
在这里插入图片描述

先序遍历

遍历过程如下:

  1. 访问根节点,
  2. 访问当前节点的左子树,
  3. 访问当前节点的右子树。

先序遍历的应用场景主要是对树进行复制。对于上面的树,先序遍历的顺序是1 2 4 5 3.
简单的用递归实现一下:

def printPreorder(root): 
  
    if root: 
        print(root.val), 
        printPreorder(root.left) 
        printPreorder(root.right) 

中序遍历

遍历过程如下:

  1. 访问左子树,
  2. 访问根节点,
  3. 访问右子树。

中序遍历的应用场景主要是在二叉搜索树上,因为返回的节点值是递增的。也可以通过对中序遍历进行变形来返回递减的BST值。在上面的例子中,中序遍历的顺序应该是4 2 5 1 3.

def printInorder(root): 
  
    if root: 
        printInorder(root.left) 
        print(root.val), 
        printInorder(root.right) 

此处应有一道leetcode算法题lol:
94. Binary Tree Inorder Traversal(https://leetcode-cn.com/problems/binary-tree-inorder-traversal/)
用栈的方法来做:

def inorderTraversal(root):
        res = []
        node_stack = []
        p = root
        while p or node_stack:
            while p:
                stack.append(p)
                p = p.left
            cur_node = node_stack.pop()
            res.append(cur_node .val)
            p = cur_node .right
        return res

还有另外一道类似的题:
95. Binary Search Tree Iterator (https://leetcode-cn.com/problems/binary-search-tree-iterator/),这道题的解决方法其实也是用栈来实现中序遍历。
因为题目相关,在这里简单提及一下迭代器的概念:

迭代是Python最强大的功能之一,是访问集合元素的一种方式。
迭代器是一个可以记住遍历的位置的对象。
迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。迭代器只能往前不会后退。
迭代器有两个基本的方法:iter() 和 next()。
字符串,列表或元组对象都可用于创建迭代器

详细定义和例子可以查看https://www.runoob.com/python3/python3-iterator-generator.html

后序遍历

遍历过程如下:

  1. 访问左子树,
  2. 访问右子树,
  3. 访问根节点。

后序遍历可以用来对树进行删除。上面的例子用后序遍历的正确顺序是 4 5 2 3 1.。

def printPostorder(root): 
    if root: 
        printPostorder(root.left) 
        printPostorder(root.right) 
        print(root.val), 

leetcode中相关的算法题:

  1. Binary Tree Postorder Traversal(https://leetcode-cn.com/problems/binary-tree-postorder-traversal/)
    解法如下:
    def postorderTraversal(root):
        if not root:
            return []
        res = []
        node_stack = [root]
        while node_stack:
            node = node_stack.pop()
            if node.left :
                node_stack.append(node.left)
            if node.right:
                node_stack.append(node.right)
            res.append(node.val)
        return res[::-1]

另外看到一个比较好的遍历方法,读起来非常清晰易理解且容易实现,来自@henry, 现在摘抄如下

在树的深度优先遍历中(包括前序、中序、后序遍历),递归方法最为直观易懂,但考虑到效率,我们通常不推荐使用递归。

栈迭代方法虽然提高了效率,但其嵌套循环却非常烧脑,不易理解,容易造成“一看就懂,一写就废”的窘况。而且对于不同的遍历顺序(前序、中序、后序),循环结构差异很大,更增加了记忆负担。

因此,我在这里介绍一种“颜色标记法”(瞎起的名字……),兼具栈迭代方法的高效,又像递归方法一样简洁易懂,更重要的是,这种方法对于前序、中序、后序遍历,能够写出完全一致的代码。

其核心思想如下:
使用颜色标记节点的状态,新节点为白色,已访问的节点为灰色。
如果遇到的节点为白色,则将其标记为灰色,然后将其右子节点、自身、左子节点依次入栈。
如果遇到的节点为灰色,则将节点的值输出。
使用这种方法实现的中序遍历如下;如要实现前序、后序遍历,只需要调整左右子节点的入栈顺序即可。

class Solution:
    def inorderTraversal(self, root: TreeNode) -> List[int]:
        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

附上原链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal/solution/yan-se-biao-ji-fa-yi-chong-tong-yong-qie-jian-ming/

您可能感兴趣的与本文相关的镜像

Python3.10

Python3.10

Conda
Python

Python 是一种高级、解释型、通用的编程语言,以其简洁易读的语法而闻名,适用于广泛的应用,包括Web开发、数据分析、人工智能和自动化脚本

1本程在vc++6.0编译通过并能正常运行。 2主界面 程已经尽量做到操作简便了,用户只需要根据提示一步步进行操作就行了。 六思考和总结: 这个课程设计的各个基本操作大部分都在我的综合性实验中实现了,所以做这个主要攻克插入和删除这两个算法!其中插入在书本上已经有了,其中的右平衡算法虽然没有给出,但通过给出的左平衡算法很容易就可以写出右平衡算法。所以最终的点就在于删除算法的实现!做的过程中对插入算法进行了非常非常多次的尝试!花了非常多的时间,这其中很多时候是在对程进行单步调试,运用了VC6。0的众多良好工具,也学到了很多它的许多好的调试手段。 其中删除算法中最难想到的一点是:在用叶子结点代替要删除的非叶子结点后,应该递归的运用删除算法去删除叶子结点!这就是整个算法的核心,其中很强烈得体会到的递归的强大,递归的最高境界(我暂时能看到的境界)! 其它的都没什么了。选做的那两个算法很容易实现的: 1合并两棵平衡二叉排树:只需遍历其中的一棵,将它的每一个元素插入到另一棵即可。 2拆分两棵平衡二叉排树:只需以根结点为中心,左子树独立为一棵,右子树独立为一棵,最后将根插入到左子树或右子树即可。 BSTreeEmpty(BSTree T) 初始条件:平衡二叉排树存在。 操作结果:若T为空平衡二叉排树,则返回TRUE,否则FALSE. BSTreeDepth(BSTree T) 初始条件:平衡二叉排树存在。 操作结果:返回T的深度。 LeafNum(BSTree T) 求叶子结点数,非递归中遍历 NodeNum(BSTree T) 求结点数,非递归中遍历 DestoryBSTree(BSTree *T) 后序遍历销毁平衡二叉排树T R_Rotate(BSTree *p) 对以*p为根的平衡二叉排树作右旋处理,处理之后p指向新的树根结点 即旋转处理之前的左子树的根结点 L_Rotate(BSTree *p) 对以*p为根的平衡二叉排树作左旋处理,处理之后p指向新的树根结点, 即旋转处理之前的右子树的根结点 LeftBalance(BSTree *T) 对以指针T所指结点为根的平衡二叉排树作左平衡旋转处理, 本算法结束时,指针T指向新的根结点 RightBalance(BSTree *T) 对以指针T所指结点为根的平衡二叉排树作右平衡旋转处理, 本算法结束时,指针T指向新的根结点 Insert_AVL(BSTree *T, TElemType e, int *taller) 若在平衡的二叉排树T中不存在和e有相同的关键字的结点, 则插入一个数据元素为e的新结点,并返回OK,否则返回ERROR. 若因插入而使二叉排树失去平衡,则作平衡旋转处理 布尔变量taller反映T长高与否 InOrderTraverse(BSTree T) 递归中遍历输出平衡二叉排树 SearchBST(BSTree T, TElemType e, BSTree *f, BSTree *p) 在根指针T所指的平衡二叉排树中递归的查找其元素值等于e的数据元素, 若查找成功,则指针p指向该数据元素结点,并返回TRUE,否则指针p 指向查找路径上访问的最后一个结点并返回FALSE,指针f指向T的双亲, 其初始调用值为NULL Delete_AVL(BSTree *T, TElemType e, int *shorter) 在平衡二叉排树中删除元素值为e的结点,成功返回OK,失败返回ERROR PrintBSTree_GList(BSTree T) 以广义表形式打印出来 PrintBSTree_AoList(BSTree T, int length) 以凹入表形式打印,length初始值为0 Combine_Two_AVL(BSTree *T1, BSTree T2) 合并两棵平衡二叉排树 Split_AVL(BSTree T, BSTree *T1, BSTree *T2) 拆分两棵平衡二叉树 } (2)存储结构的定义: typedef struct BSTNode { TElemType data; int bf; //结点的平衡因子 struct BSTNode *lchild, *rchild;//左.右孩子指针 }BSTNode, *BSTree;
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值