码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 94.二叉树的中序遍历

时间:2020-07-19 11:38:09      阅读:55      评论:0      收藏:0      [点我收藏+]

标签:ack   lan   def   遍历   null   pre   one   完成   子节点   

给定一个二叉树,返回它的中序 遍历。
示例:
输入: [1,null,2,3]
   1
         2
    /
   3

输出: [1,3,2]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
## 迭代
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

## 递归
class Solution:
    def inorderTraversal(self, root: TreeNode) -> List[int]:
        res = []
        self.helper(root, res)
        return res

    def helper(self, root, res):
        if root:
            self.helper(root.left, res)
            res.append(root.val)
            self.helper(root.right, res)


LeetCode 94.二叉树的中序遍历

标签:ack   lan   def   遍历   null   pre   one   完成   子节点   

原文地址:https://www.cnblogs.com/sandy-t/p/13338804.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!