来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
给定一个二叉树的根节点 root ,返回 它的 中序 遍历 。
输入:root = [1,null,2,3]
输出:[1,3,2]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
提示:
树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
对根来说的前中后
前序(根左右),中序(左根右),后序(左右根)
(1)递归
链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal/solution/er-cha-shu-de-zhong-xu-bian-li-by-leetcode-solutio/
因此整个遍历过程天然具有递归的性质,我们可以直接用递归函数来模拟这一过程。
class Solution:
def inorderTraversal(self, root: TreeNode) -> List[int]:
ret = []
if root:
ret += self.inorderTraversal(root.left)
ret.append(root.val)
ret += self.inorderTraversal(root.right)
return ret
复杂度分析
时间复杂度:O(n)O(n),其中 nn 为二叉树节点的个数。二叉树的遍历中每个节点会被访问一次且只会被访问一次。
空间复杂度:O(n)O(n)。空间复杂度取决于递归的栈深度,而栈深度在二叉树为一条链的情况下会达到 O(n)O(n) 的级别。
本地可运行的
from typing import List, Optional
# 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: TreeNode) -> List[int]:
ret = []
if root:
ret += self.inorderTraversal(root.left)
ret.append(root.val)
ret += self.inorderTraversal(root.right)
return ret
l1=TreeNode(1)
l1.right=TreeNode(2)
l1.right.left=TreeNode(3)
aa = Solution()
bb = aa.inorderTraversal(l1)
print(bb)
(2)迭代
链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal/solution/er-cha-shu-de-zhong-xu-bian-li-by-leetcode-solutio/
方法一的递归函数我们也可以用迭代的方式实现,两种方式是等价的,区别在于递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来,其他都相同,具体实现可以看下面的代码。
class Solution:
def inorderTraversal(self, root: TreeNode) -> List[int]:
def add_all_left(node):#添加所有的左节点到栈中
while node:
stack.append(node)
node = node.left
stack, res = [], []
add_all_left(root)#从根节点,将所有左节点加到栈中
while stack:
cur = stack.pop()#出栈
res.append(cur.val)#添加当前值到结果
add_all_left(cur.right)#对右节点进行添加所有的左节点
return res
复杂度分析
时间复杂度:O(n)O(n),其中 nn 为二叉树节点的个数。二叉树的遍历中每个节点会被访问一次且只会被访问一次。
空间复杂度:O(n)O(n)。空间复杂度取决于栈深度,而栈深度在二叉树为一条链的情况下会达到 O(n)O(n) 的级别。
(3)Morris (莫里斯)中序遍历—把空间复杂度优化到O(1)的二叉树遍历算法。
class Solution:
def inorderTraversal(self, root: TreeNode) -> List[int]:
res = []
while root:
if root.left:
# find out predecessor
predecessor = root.left
while predecessor.right:
predecessor = predecessor.right
# link predecessor to root
predecessor.right = root
# set left child of root to None
temp = root
root = root.left
temp.left = None
else: