题目描述:
给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
个人思路:
这道题和590. N叉树的后序遍历思路基本一致,不同之处在于N叉树会提供当前节点的所有孩子节点,而二叉树则是通过左孩子和右孩子指针给出。还是一样,只需要注意以下两点即可:
- 孩子节点之间需要按照从左到右的自然顺序完成遍历
- 父亲节点要晚于所有孩子节点完成遍历
具体代码如下:
# 递归
class Solution:
def postorderTraversal(self, root: TreeNode) -> List[int]:
if not root:
return []
results = []
def helper(root):
if not root:
return
self.postorderTraversal(root.left) # L
self.postorderTraversal(root.right) # R
results.append(root.val) # D
helper(root)
return results
# 迭代
class Solution:
def postorderTraversal(self, root: TreeNode) -> List[int]:
if not root:
return []
results = []
stack = [root]
while stack:
fathers = stack
while fathers: # 和N叉树不同之处在于,还需要通过一次循环才能获取当前父节点的所有孩子
father = fathers.pop()
if father:
results.append(father) # 先遍历父节点
# 顺序添加左右孩子,再后续通过下一次遍历中的pop操作完成反序
stack.append(father.left)
stack.append(father.right)
return results[::-1]