二叉树的后序遍历
难度:简单
给你一棵二叉树的根节点 root ,返回其节点值的后序遍历 。
示例 1:
输入:root = [1,null,2,3]
输出:[3,2,1]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
Morris 遍历
思路:
有一种巧妙的方法可以在线性时间内,只占用常数空间来实现后序遍历。这种方法由 J. H. Morris 在 1979 年的论文「Traversing Binary Trees Simply and Cheaply」中首次提出,因此被称为 Morris 遍历。
Morris 遍历的核心思想是利用树的大量空闲指针,实现空间开销的极限缩减。其后序遍历规则总结如下:
- 新建临时节点,令该节点为 root;
- 如果当前节点的左子节点为空,则遍历当前节点的右子节点;
- 如果当前节点的左子节点不为空,在当前节点的左子树中找到当前节点在中序遍历下的前驱节点;
- 如果前驱节点的右子节点为空,将前驱节点的右子节点设置为当前节点,当前节点更新为当前节点的左子节点。
- 如果前驱节点的右子节点为当前节点,将它的右子节点重新设为空。倒序输出从当前节点的左子节点到该前驱节点这条路径上的所有节点。当前节点更新为当前节点的右子节点。
- 重复步骤 2 和步骤 3,直到遍历结束。
这样我们利用 Morris 遍历的方法,后序遍历该二叉搜索树,即可实现线性时间与常数空间的遍历。
时间复杂度:
O
(
n
)
O(n)
O(n),其中 n 是二叉树的节点数。没有左子树的节点只被访问一次,有左子树的节点被访问两次。
空间复杂度:
O
(
1
)
O(1)
O(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 reversal(self, node):
prev = None
while node:
temp = node.right
node.right = prev
prev = node
node = temp
return prev
def append_node(self, node, lis):
node = self.reversal(node)
while node:
lis.append(node.val)
node = node.right
self.reversal(node)
def postorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
res = []
node = root
while root:
most_right = root.left
if most_right:
while most_right.right and most_right.right != root:
most_right = most_right.right
if not most_right.right:
most_right.right = root
root = root.left
continue
else:
most_right.right = None
self.append_node(root.left, res)
root = root.right
self.append_node(node, res)
return res
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/binary-tree-postorder-traversal