94 - 二叉树的中序遍历 - python

384 阅读1分钟

给定一个二叉树,返回它的中序 遍历。

示例:

输入: [1,null,2,3]
   1
    \
     2
    /
   3

输出: [1,3,2]

进阶: 递归算法很简单,你可以通过迭代算法完成吗?


二叉树的中序遍历顺序为:左- 根 - 右,最常用的是使用递归来进行求解,递归过程是:遍历左子树 - 打印当前结点 - 遍历右子树 ,递归终止条件是root == None

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

        if root == []: return [] 

        global res
        res = []
        inOrder(root)

        return res

题目要求使用迭代法求解二叉树的中序遍历,树的遍历顺序仍然要遵从左-根-右的原则。因此,迭代法使用栈来保存遍历过的结点。从根节点出发先遍历左子树,一直往下走直到为叶子结点,然后将当前结点出栈,然后判断是否右右结点,如果有则继续出栈,重复此过程直到栈为空,表示已到达根节点。右子树重复前面的过程,直到栈为空。
在这里插入图片描述

class Solution:
    def inorderTraversal(self, root: TreeNode) -> List[int]:
        if root == None: return []
        res = []
        stack = []
        while stack or root:
            if root:
                stack.append(root)
                root = root.left
            else:
                node = stack.pop()
                res.append(node.val)
                root = node.right
        
        return res

还有一种改变树结构的中序遍历方式-莫里斯遍历,有兴趣的可查阅官方解答或查阅相关资料