1.题目描述
给你二叉树的根节点
root
,返回它节点值的 中序遍历。
输入:root = [1,null,2,3]
输出:[1,3,2]输入:root = []
输出:[]输入:root = [1]
输出:[1]提示:
- 树中节点数目在范围
[0, 100]
内-100 <= Node.val <= 100
2.思路分析
二叉树的前序遍历: 按 按照访问左子树——根节点——右子树的方式遍历这棵树
2.1 递归
- 递归调用
inorder(root.left)
来遍历root
节点的左子树 - 将
root
节点的值加入答案 - 递归调用
inorder(root.right)
来遍历root
节点的右子树
2.2 迭代
递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来
迭代法写中序遍历,就需要借用指针的遍历来帮助访问节点,栈则用来处理节点上的元素。
3.代码实现
3.1 递归法
# 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]:
# 深度优先遍历 中序遍历:左--中--右
# 定义保存结果的数组
result = []
def traversal(root: Optional[TreeNode]) -> None:
if not root:
return
traversal(root.left) # 左
result.append(root.val) # 中
traversal(root.right) # 右
traversal(root)
return result
- 时间复杂度:O(n),其中 n 是二叉树的节点数。每一个节点恰好被遍历一次。
- 空间复杂度:O(n),为递归过程中栈的开销,平均情况下为 O(logn),最坏情况下树呈现链状,为 O(n)。
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]:
result = []
if not root:
return result
# # 访问和处理的元素不同 指针:访问元素 栈:处理结果
cur = root
stack = []
while stack or cur:
if cur:
stack.append(cur)
cur = cur.left
else:
cur = stack.pop()
result.append(cur.val)
cur = cur.right
return result
- 时间复杂度:O(n),其中 n 是二叉树的节点数。每一个节点恰好被遍历一次。
- 空间复杂度:O(n),为迭代过程中显式栈的开销,平均情况下为O(logn),最坏情况下树呈现链状,为 O(n)。