目录
一,题目描述
英文描述
Given the root of a binary tree, return the postorder traversal of its nodes' values.
中文描述
给定一个二叉树,返回它的 后序 遍历。
示例与说明
Example 1:
Example 2:Example 3:
Example 4:
Example 5:Constraints:
The number of the nodes in the tree is in the range [0, 100].
-100 <= Node.val <= 100
Follow up: Recursive solution is trivial, could you do it iteratively?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二,解题思路
参考官方题解@力扣官方题解【 二叉树的后序遍历】
关键在于用栈模拟递归的过程,包括递归的进度、递归的状态。
- 递归的进度:标明遍历到了哪个节点;
- 递归的状态:标明左右子树是否已遍历;
只要能记录上面两种信息,不管什么方法都可以。
官方题解中采用栈stack来记录递归的进度,prev节点记录递归的状态。
- stack:遵循先左后右的规则将节点入栈,每输出一个值到结果数组中时,将该节点从栈里彻底弹出(不再入栈);
- prev:记录遍历的上一节点,用来判断右子树是否已经遍历(以下图为例,遍历完节点7后,将其标记为prev。当遍历到下一个节点4时,凭借root.right == prev?来判断右子树是否已遍历);
三,AC代码
Java
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> ans = new ArrayList<>();
if (root == null) return ans;
Deque<TreeNode> stack = new LinkedList<>();
TreeNode pre = null;
while (root != null || !stack.isEmpty()) {
// 一路向左 左节点入栈
while (root != null) {
stack.push(root);
root = root.left;
}
root = stack.pop();
// 到这里时 已经可以表示左子树为空或者左子树已经全部遍历过
// 叶子节点、右节点为空或者右子树已遍历 将节点值存入结果数组
// pre = root 将来用于判断右子树是否已遍历
if (root.right == null || root.right == pre) {
ans.add(root.val);
pre = root;
root = null;
} else {
// 根节点重新入栈(因为根节点的值还没有输出到结果数组中)
// 并将右子树节点调整为新根节点 开始新子树的迭代
stack.push(root);
root = root.right;
}
}
return ans;
}
}
四,解题过程
第一博
递归方法大家都会,迭代方法实现的话,这一题难度可以上升到medium甚至hard。直接参考了官方题解。
关键在于用栈模拟递归的过程,包括递归的进度、递归的状态。