【力扣】94. 二叉树的中序遍历(递归、栈)

最新推荐文章于 2024-09-13 18:11:35 发布

数码宝贝饲养员

最新推荐文章于 2024-09-13 18:11:35 发布

阅读量261

点赞数

分类专栏：刷题文章标签： leetcode

本文链接：https://blog.csdn.net/qq_42245490/article/details/105906905

版权

刷题专栏收录该内容

11 篇文章 0 订阅

订阅专栏

本文介绍了如何解决力扣94题——二叉树的中序遍历，主要讲解了递归和颜色标记法两种解题思路。颜色标记法通过标记节点状态辅助遍历，当遇到白色节点时将其标记为灰色并入栈，遇到灰色节点则输出其值。详细解析及代码实现见原文。

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

【力扣】94. 二叉树的中序遍历

递归方法：

class Solution:
    def inorderTraversal(self, root: TreeNode) -> List[int]:
        if not root:
            return []
        val = root.val
        left = self.inorderTraversal(root.left)
        right = self.inorderTraversal(root.right)
        return left + [val] + right

颜色标记法：
其核心思想如下：
使用颜色标记节点的状态，新节点为白色，已访问的节点为灰色。
如果遇到的节点为白色，则将其标记为灰色，然后将其右子节点、自身、左子节点依次入栈。
如果遇到的节点为灰色，则将节点的值输出。

作者：hzhu212
链接：https://leetcode-cn.com/problems/binary-tree-inorder-traversal/solution/yan-se-biao-ji-fa-yi-chong-tong-yong-qie-jian-ming/
来源：力扣（LeetCode）
代码：

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