LeeCode题目：二叉树中序遍历

题目描述

给定一棵二叉树的根节点，返回树中节点值的中序遍历。

示例

输入： root = [1,null,2,3]
输出： [1,3,2]

回顾二叉树遍历

三种遍历方式，遍历顺序其实是相对于根节点
前序遍历：根节点->左子树->右子树
中序遍历：左子树->根节点->右子树
后序遍历：左子树->右子树->根节点

代码实现

// 使用递归算法实现
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        if root  is None:
            return []
        res = []
        left_nodes = self.inorderTraversal(root.left)
        res.extend(left_nodes)
        res.append(root.val)
        right_nodes = self.inorderTraversal(root.right)
        res.extend(right_nodes)
        return res

虽然递归算法实现简单，但是其空间和时间复杂度较高，写程序时要尽量避免使用递归

// 使用迭代算法实现
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        # 利用栈来进行二叉树的中序遍历
        # 定义空的返回列表和栈
        res, stack = [], []
        while True:
        # 当根节点不为空时，将根节点与它的连续左节点压入栈中
            while root:
                stack.append(root)
                root = root.left
            if not stack:
                return res
            # 遍历左子树和根节点的值
            node = stack.pop()
            res.append(node.val)
            # 遍历右子树
            root = node.right